Spaces:
Runtime error
Runtime error
Update requirements.txt
Browse files- requirements.txt +4 -0
requirements.txt
CHANGED
|
@@ -1,3 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
accelerate==0.16.0
|
| 2 |
aiohttp==3.8.4
|
| 3 |
aiosignal==1.3.1
|
|
|
|
| 1 |
+
torch==2.0.0+cu117 --index-url https://download.pytorch.org/whl/cu117
|
| 2 |
+
torchvision==0.15.1+cu117 --index-url https://download.pytorch.org/whl/cu117
|
| 3 |
+
torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
|
| 4 |
+
# CUDA 11.8
|
| 5 |
accelerate==0.16.0
|
| 6 |
aiohttp==3.8.4
|
| 7 |
aiosignal==1.3.1
|