Spaces:
Running
Running
Update requirements.txt
Browse files- requirements.txt +11 -12
requirements.txt
CHANGED
@@ -1,10 +1,13 @@
|
|
1 |
# Core dependencies
|
2 |
torch>=2.0.0
|
3 |
-
transformers>=4.
|
4 |
accelerate>=0.21.0
|
5 |
bitsandbytes>=0.41.0
|
6 |
sentencepiece>=0.1.99
|
7 |
protobuf>=3.20.0
|
|
|
|
|
|
|
8 |
|
9 |
# Multimodal processing
|
10 |
pillow>=9.5.0
|
@@ -18,22 +21,18 @@ gradio>=3.50.2
|
|
18 |
sentence-transformers>=2.2.2
|
19 |
faiss-cpu>=1.7.4
|
20 |
|
|
|
|
|
|
|
21 |
# Utilities
|
22 |
numpy>=1.24.0
|
23 |
scikit-learn>=1.2.2
|
24 |
tqdm>=4.65.0
|
25 |
-
|
26 |
-
# Better compatibility with Hugging Face Spaces
|
27 |
huggingface-hub>=0.19.0
|
28 |
-
safetensors>=0.4.0
|
29 |
|
30 |
-
# Optional: improved image
|
31 |
torchvision>=0.15.0
|
32 |
|
33 |
-
#
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
# Qwen specific requirements
|
38 |
-
tiktoken>=0.4.0
|
39 |
-
einops>=0.6.1
|
|
|
1 |
# Core dependencies
|
2 |
torch>=2.0.0
|
3 |
+
transformers>=4.36.0
|
4 |
accelerate>=0.21.0
|
5 |
bitsandbytes>=0.41.0
|
6 |
sentencepiece>=0.1.99
|
7 |
protobuf>=3.20.0
|
8 |
+
einops>=0.6.1
|
9 |
+
tiktoken>=0.4.0
|
10 |
+
safetensors>=0.4.0
|
11 |
|
12 |
# Multimodal processing
|
13 |
pillow>=9.5.0
|
|
|
21 |
sentence-transformers>=2.2.2
|
22 |
faiss-cpu>=1.7.4
|
23 |
|
24 |
+
# Optional: use faiss with GPU (if CUDA available)
|
25 |
+
# faiss-gpu>=1.7.4
|
26 |
+
|
27 |
# Utilities
|
28 |
numpy>=1.24.0
|
29 |
scikit-learn>=1.2.2
|
30 |
tqdm>=4.65.0
|
|
|
|
|
31 |
huggingface-hub>=0.19.0
|
|
|
32 |
|
33 |
+
# Optional: improved image preprocessing
|
34 |
torchvision>=0.15.0
|
35 |
|
36 |
+
# Additional for robustness
|
37 |
+
requests>=2.31.0
|
38 |
+
typing-extensions>=4.5.0
|
|
|
|
|
|
|
|