jawakja commited on
Commit
d54883e
·
verified ·
1 Parent(s): 03f20a2

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -12
requirements.txt CHANGED
@@ -1,10 +1,13 @@
1
  # Core dependencies
2
  torch>=2.0.0
3
- transformers>=4.31.0
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 processing
31
  torchvision>=0.15.0
32
 
33
- # Optional: CUDA support for faiss (if GPU available)
34
- # Uncomment the line below if you have GPU access
35
- # faiss-gpu>=1.7.4
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