#!/bin/bash
set -e

# Clona il repository
git clone https://github.com/hpcaitech/Open-Sora
cd Open-Sora
pip install -v -e .
cd ..

# Installa xformers con l'esatta versione specificata
pip install xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121

# Installa flash-attn senza build isolation
pip install flash-attn --no-build-isolation

# Scarica il modello
mkdir -p ./ckpts
huggingface-cli download hpcai-tech/Open-Sora-v2 --local-dir ./ckpts