Update requirements.txt
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
seaborn
|
| 2 |
scipy
|
| 3 |
-
pytorch
|
| 4 |
-
pytorch-cuda
|
| 5 |
datasets==2.14.5
|
| 6 |
gradio
|
| 7 |
huggingface-hub==0.18.0
|
|
|
|
| 1 |
seaborn
|
| 2 |
scipy
|
| 3 |
+
pytorch==2
|
| 4 |
+
pytorch-cuda==12.1
|
| 5 |
datasets==2.14.5
|
| 6 |
gradio
|
| 7 |
huggingface-hub==0.18.0
|