Spaces:
Paused
Paused
Update requirements.txt
Browse files- requirements.txt +4 -4
requirements.txt
CHANGED
@@ -31,12 +31,12 @@ termcolor==2.2.0
|
|
31 |
thinc==8.1.10
|
32 |
timm==0.6.12
|
33 |
tokenizers==0.13.2
|
34 |
-
--extra-index-url https://download.pytorch.org/whl/
|
35 |
-
torch==1.13.1+
|
36 |
-
torchvision==0.14.1+
|
37 |
tqdm==4.64.1
|
38 |
transformers==4.26.1
|
39 |
triton
|
40 |
wandb==0.13.10
|
41 |
-
xformers==0.0.
|
42 |
modelcards
|
|
|
31 |
thinc==8.1.10
|
32 |
timm==0.6.12
|
33 |
tokenizers==0.13.2
|
34 |
+
--extra-index-url https://download.pytorch.org/whl/cu117
|
35 |
+
torch==1.13.1+cu117
|
36 |
+
torchvision==0.14.1+cu117
|
37 |
tqdm==4.64.1
|
38 |
transformers==4.26.1
|
39 |
triton
|
40 |
wandb==0.13.10
|
41 |
+
xformers==0.0.16
|
42 |
modelcards
|