Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
|
@@ -1,7 +1,8 @@
|
|
| 1 |
-
--extra-index-url https://download.pytorch.org/whl/
|
|
|
|
| 2 |
gradio==5.33.1
|
| 3 |
-
torch==2.1.0
|
| 4 |
-
torchvision==0.16.0+
|
| 5 |
transformers==4.44.0
|
| 6 |
accelerate==0.24.1
|
| 7 |
pillow==10.0.1
|
|
@@ -20,5 +21,4 @@ retina-face==0.0.13
|
|
| 20 |
gdown==4.7.1
|
| 21 |
tqdm==4.66.1
|
| 22 |
|
| 23 |
-
# Flash Attention - use prebuilt wheel if available
|
| 24 |
flash-attn==2.7.4.post1 --no-build-isolation
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 2 |
+
|
| 3 |
gradio==5.33.1
|
| 4 |
+
torch==2.1.0+cu118
|
| 5 |
+
torchvision==0.16.0+cu118
|
| 6 |
transformers==4.44.0
|
| 7 |
accelerate==0.24.1
|
| 8 |
pillow==10.0.1
|
|
|
|
| 21 |
gdown==4.7.1
|
| 22 |
tqdm==4.66.1
|
| 23 |
|
|
|
|
| 24 |
flash-attn==2.7.4.post1 --no-build-isolation
|