Spaces:
Runtime error
Runtime error
erwold
commited on
Commit
·
3968707
1
Parent(s):
fa28a3d
Initial Commit
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
# Core dependencies
|
| 2 |
-
|
| 3 |
-
|
|
|
|
| 4 |
transformers==4.45.0
|
| 5 |
diffusers==0.30.0
|
| 6 |
accelerate==0.33.0
|
|
@@ -13,5 +14,4 @@ numpy>=1.24.0
|
|
| 13 |
# Utilities
|
| 14 |
protobuf==4.23.4
|
| 15 |
sentencepiece==0.2.0
|
| 16 |
-
|
| 17 |
gradio==4.36.0
|
|
|
|
| 1 |
# Core dependencies
|
| 2 |
+
--extra-index-url https://download.pytorch.org/whl/cu118
|
| 3 |
+
torch==2.4.1+cu118
|
| 4 |
+
torchvision==0.15.0+cu118
|
| 5 |
transformers==4.45.0
|
| 6 |
diffusers==0.30.0
|
| 7 |
accelerate==0.33.0
|
|
|
|
| 14 |
# Utilities
|
| 15 |
protobuf==4.23.4
|
| 16 |
sentencepiece==0.2.0
|
|
|
|
| 17 |
gradio==4.36.0
|