stephenebert commited on
Commit
941245a
Β·
verified Β·
1 Parent(s): 3361a4d

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +21 -13
requirements.txt CHANGED
@@ -1,13 +1,21 @@
1
- # core UI libs
2
- gradio==3.50.2
3
- click<8.1
4
- pydantic>=1.10.0
5
-
6
- # vision / embedding stack
7
- torch==2.6.0+cpu
8
- torchvision==0.19.0+cpu
9
- torchaudio==2.6.0+cpu # optional; delete if unused
10
- -f https://download.pytorch.org/whl/torch_stable.html
11
-
12
- sentence-transformers==2.5.1 # β‰₯2.5.x ships wheels for Py3.10+
13
- faiss-cpu==1.7.4
 
 
 
 
 
 
 
 
 
1
+ # ------------- Core deep-learning stack (CPU wheels) -------------
2
+ torch==2.6.0 # plain tag β†’ CPU wheel; no "+cpu" suffix
3
+ torchvision==0.19.0 # matches the Torch major/minor
4
+
5
+ # ------------- Transformers / BLIP / CLIP ------------------------
6
+ transformers==4.42.0 # recent HF Transformers
7
+ sentence-transformers==2.6.1
8
+ timm==0.9.16 # visual backbones used by BLIP/CLIP
9
+ accelerate==0.29.3 # hf accelerate helper
10
+
11
+ # ------------- Vector search -------------------------------------
12
+ faiss-cpu==1.8.0 # FAISS with CPU SIMD
13
+
14
+ # ------------- Gradio UI -----------------------------------------
15
+ gradio==3.50.2 # last 3.x release compatible with Spaces
16
+
17
+ # ------------- Misc utilities ------------------------------------
18
+ numpy>=1.23,<1.27
19
+ pillow>=10.0
20
+ tqdm>=4.66
21
+