Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
f9b4029
1
Parent(s):
aeea06f
Update requirements.txt
Browse files- requirements.txt +4 -1
requirements.txt
CHANGED
|
@@ -1,5 +1,9 @@
|
|
|
|
|
|
|
|
| 1 |
torch==2.4.0
|
| 2 |
torchvision==0.19.0
|
|
|
|
|
|
|
| 3 |
pytorch-lightning==2.4.0
|
| 4 |
einops
|
| 5 |
plotly
|
|
@@ -12,7 +16,6 @@ transformers==4.46.1
|
|
| 12 |
tokenizers==0.20.1
|
| 13 |
diffusers==0.31.0
|
| 14 |
accelerate==1.0.1
|
| 15 |
-
xformers===0.0.27.post2
|
| 16 |
grpcio-status==1.33.2
|
| 17 |
protobuf==3.20.2
|
| 18 |
safetensors
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
|
| 3 |
torch==2.4.0
|
| 4 |
torchvision==0.19.0
|
| 5 |
+
xformers===0.0.27.post2
|
| 6 |
+
|
| 7 |
pytorch-lightning==2.4.0
|
| 8 |
einops
|
| 9 |
plotly
|
|
|
|
| 16 |
tokenizers==0.20.1
|
| 17 |
diffusers==0.31.0
|
| 18 |
accelerate==1.0.1
|
|
|
|
| 19 |
grpcio-status==1.33.2
|
| 20 |
protobuf==3.20.2
|
| 21 |
safetensors
|