Update requirements.txt
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
-
torch==2.3.1
|
2 |
-
torchvision==0.18.1
|
3 |
transformers==4.41.2
|
4 |
gradio==4.31.5
|
5 |
Pillow==10.3.0
|
6 |
-
numpy==1.26.4
|
|
|
1 |
+
torch==2.3.1 -f https://download.pytorch.org/whl/cu118/torch_stable.html
|
2 |
+
torchvision==0.18.1 -f https://download.pytorch.org/whl/cu118/torch_stable.html
|
3 |
transformers==4.41.2
|
4 |
gradio==4.31.5
|
5 |
Pillow==10.3.0
|
6 |
+
numpy==1.26.4
|