Spaces:
Running
Running
Create setup.sh
Browse files
setup.sh
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
#!/bin/bash
|
2 |
+
set -e
|
3 |
+
|
4 |
+
# Installa dipendenze specifiche
|
5 |
+
pip install transformers==4.30.2 diffusers==0.18.2
|
6 |
+
pip install xformers==0.0.20 --index-url https://download.pytorch.org/whl/cu117
|
7 |
+
|
8 |
+
# Clona Open-Sora
|
9 |
+
git clone https://github.com/hpcaitech/Open-Sora
|
10 |
+
cd Open-Sora
|
11 |
+
pip install -e .
|
12 |
+
cd ..
|
13 |
+
|
14 |
+
# Installa colossalai
|
15 |
+
pip install colossalai==0.3.0
|
16 |
+
|
17 |
+
# Scarica il modello
|
18 |
+
mkdir -p ./ckpts
|
19 |
+
huggingface-cli download hpcai-tech/Open-Sora-v2 --local-dir ./ckpts
|
20 |
+
|
21 |
+
echo "Setup completato!"
|