Update setup.sh

setup.sh

@@ -1,21 +1,18 @@
 #!/bin/bash
 set -e
 
-# Installa dipendenze specifiche
-pip install transformers==4.30.2 diffusers==0.18.2
-pip install xformers==0.0.20 --index-url https://download.pytorch.org/whl/cu117
-
-# Clona Open-Sora
+# Clona il repository
 git clone https://github.com/hpcaitech/Open-Sora
 cd Open-Sora
-pip install -e .
+pip install -v -e .
 cd ..
 
-# Installa colossalai
-pip install colossalai==0.3.0
+# 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
-
-echo "Setup completato!"
+huggingface-cli download hpcai-tech/Open-Sora-v2 --local-dir ./ckpts