Spaces:
Paused
Paused
Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
|
@@ -31,8 +31,9 @@ termcolor==2.2.0
|
|
| 31 |
thinc==8.1.10
|
| 32 |
timm==0.6.12
|
| 33 |
tokenizers==0.13.2
|
| 34 |
-
|
| 35 |
-
|
|
|
|
| 36 |
tqdm==4.64.1
|
| 37 |
transformers==4.26.1
|
| 38 |
triton
|
|
|
|
| 31 |
thinc==8.1.10
|
| 32 |
timm==0.6.12
|
| 33 |
tokenizers==0.13.2
|
| 34 |
+
--extra-index-url https://download.pytorch.org/whl/cu116
|
| 35 |
+
torch=1.13.1+cu116
|
| 36 |
+
torchvision=0.14.1+cu116
|
| 37 |
tqdm==4.64.1
|
| 38 |
transformers==4.26.1
|
| 39 |
triton
|