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