Spaces:
Running
on
Zero
Running
on
Zero
add support for cuda
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
torch==2.2.2
|
| 2 |
transformers==4.41.2
|
|
|
|
| 3 |
gradio==4.26.0
|
| 4 |
tiktoken==0.7.0
|
| 5 |
openai==1.61.0
|
|
|
|
| 1 |
torch==2.2.2
|
| 2 |
transformers==4.41.2
|
| 3 |
+
accelerate==0.30.0
|
| 4 |
gradio==4.26.0
|
| 5 |
tiktoken==0.7.0
|
| 6 |
openai==1.61.0
|