Update requirements
Browse files- requirements +2 -4
requirements
CHANGED
|
@@ -1,8 +1,6 @@
|
|
| 1 |
-
torch>=2.
|
| 2 |
transformers>=4.40.0
|
| 3 |
datasets>=2.18.0
|
| 4 |
sentence-transformers>=2.6.0
|
| 5 |
gradio>=4.24.0
|
| 6 |
-
scikit-learn>=1.4.0
|
| 7 |
-
-torch
|
| 8 |
-
+torch==2.2.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu
|
|
|
|
| 1 |
+
torch>=2.2.1
|
| 2 |
transformers>=4.40.0
|
| 3 |
datasets>=2.18.0
|
| 4 |
sentence-transformers>=2.6.0
|
| 5 |
gradio>=4.24.0
|
| 6 |
+
scikit-learn>=1.4.0
|
|
|
|
|
|