Spaces:
Build error
Build error
Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
transformers==4.44.2
|
2 |
-
torch==2.4.1
|
3 |
fastapi==0.115.0
|
4 |
uvicorn==0.30.6
|
5 |
gradio==4.44.0
|
|
|
1 |
transformers==4.44.2
|
2 |
+
torch==2.4.1+cu121
|
3 |
fastapi==0.115.0
|
4 |
uvicorn==0.30.6
|
5 |
gradio==4.44.0
|