Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
gradio==3.28.0
|
2 |
-
torch==2.1.0
|
3 |
-
transformers==4.30.0
|
4 |
-
safetensors==0.3.0
|
|
|
1 |
+
gradio==3.28.0
|
2 |
+
torch==2.1.0 # Or use 'torch==2.1.0+cu118' for CUDA-enabled GPU support
|
3 |
+
transformers==4.30.0
|
4 |
+
safetensors==0.3.0
|