Update requirements.txt
Browse files- requirements.txt +5 -3
requirements.txt
CHANGED
|
@@ -7,11 +7,11 @@ scipy==1.11.1
|
|
| 7 |
timm>=0.9.2
|
| 8 |
tokenizers
|
| 9 |
|
| 10 |
-
torch
|
| 11 |
-
torchaudio
|
| 12 |
torchdata==0.6.1
|
| 13 |
torchmetrics>=1.0.1
|
| 14 |
-
torchvision
|
| 15 |
|
| 16 |
tqdm>=4.65.0
|
| 17 |
transformers==4.35.2
|
|
@@ -27,3 +27,5 @@ dominate
|
|
| 27 |
diffusers==0.25.1
|
| 28 |
gradio==5.4.0
|
| 29 |
huggingface_hub==0.25.2
|
|
|
|
|
|
|
|
|
| 7 |
timm>=0.9.2
|
| 8 |
tokenizers
|
| 9 |
|
| 10 |
+
torch==2.2.2+cu121
|
| 11 |
+
torchaudio==2.2.2+cu121
|
| 12 |
torchdata==0.6.1
|
| 13 |
torchmetrics>=1.0.1
|
| 14 |
+
torchvision==0.17.2+cu121
|
| 15 |
|
| 16 |
tqdm>=4.65.0
|
| 17 |
transformers==4.35.2
|
|
|
|
| 27 |
diffusers==0.25.1
|
| 28 |
gradio==5.4.0
|
| 29 |
huggingface_hub==0.25.2
|
| 30 |
+
|
| 31 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|