Ritvik Shrivastava commited on
Commit
cc17576
Β·
1 Parent(s): 41c0a9e

fix: compatible torch version for ZeroGPU

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -31
requirements.txt CHANGED
@@ -1,32 +1,12 @@
1
- # ── Core ──────────────────────────────────────────────────────────
2
- gradio>=6.14.0 # gr.Server custom routing (strict minimum)
3
- fastapi>=0.111.0 # gr.Server extends FastAPI
4
- uvicorn>=0.29.0
5
-
6
- # ── Model / Inference ─────────────────────────────────────────────
7
- torch==2.1.2
8
- torchvision==0.16.2
9
- transformers==4.40.0
10
- peft==0.10.0 # LoRA adapter loading
11
- accelerate==0.27.0
12
- sentencepiece==0.1.99 # MiniCPM-V tokenizer
13
- Pillow==10.1.0
14
-
15
- # ── HuggingFace Ecosystem ─────────────────────────────────────────
16
- huggingface_hub>=0.22.0
17
- datasets>=2.18.0
18
- spaces>=0.28.0 # @spaces.GPU decorator for ZeroGPU
19
-
20
- # ── Dataset Pipeline (scripts only) ──────────────────────────────
21
- requests>=2.31.0
22
- beautifulsoup4>=4.12.0
23
- selenium>=4.18.0 # Tier-2 scraping
24
- open-clip-torch>=2.24.0 # CLIP deduplication
25
- tqdm>=4.66.0
26
- pandas>=2.2.0
27
- scikit-learn>=1.4.0
28
-
29
- # ── Utilities ─────────────────────────────────────────────────────
30
- python-dotenv>=1.0.0
31
  loguru>=0.7.0
32
- numpy>=1.26.0
 
 
 
 
1
+ gradio>=6.14.0
2
+ transformers==4.45.2
3
+ peft==0.10.0
4
+ torch==2.2.2
5
+ Pillow>=10.0.0
6
+ huggingface_hub>=0.36.0
7
+ qwen-vl-utils>=0.0.8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  loguru>=0.7.0
9
+ spaces>=0.28.0
10
+ fastapi>=0.111.0
11
+ uvicorn>=0.29.0
12
+ sentencepiece==0.1.99