Update requirements.txt
Browse files- requirements.txt +3 -2
requirements.txt
CHANGED
@@ -1,7 +1,8 @@
|
|
1 |
gradio>=3.0
|
2 |
requests
|
3 |
-
sentence-transformers
|
4 |
scikit-learn
|
5 |
numpy
|
6 |
transformers==4.30.0
|
7 |
-
torch
|
|
|
|
1 |
gradio>=3.0
|
2 |
requests
|
3 |
+
sentence-transformers==2.2.2 # Compatible with transformers==4.30.0 and older huggingface_hub
|
4 |
scikit-learn
|
5 |
numpy
|
6 |
transformers==4.30.0
|
7 |
+
torch==2.0.1 # Matches transformers 4.30.0
|
8 |
+
huggingface-hub==0.25.2 # Matches transformers 4.30.0 and avoids cached_download removal
|