RishiRP commited on
Commit
ec2fccc
·
verified ·
1 Parent(s): 89f7ef2

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -26
requirements.txt CHANGED
@@ -1,26 +1,18 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
2
-
3
- # PyTorch (CUDA 11.8)
4
- torch==2.2.2+cu118
5
- torchvision==0.17.2+cu118
6
- torchaudio==2.2.2+cu118
7
-
8
- # Core inference
9
- transformers==4.43.3
10
- accelerate==0.33.0
11
- safetensors==0.4.4
12
- bitsandbytes==0.43.1
13
-
14
- # UI & client (match Gradio 4.44.0 requirements)
15
- gradio==4.44.0
16
- gradio_client==1.3.0
17
- websockets>=12.0 # required for websockets.asyncio.*
18
-
19
- # Data / utils
20
- pandas==2.2.2
21
- numpy==1.26.4
22
- regex==2024.7.24
23
- huggingface_hub==0.24.6
24
-
25
- # Tokenization / sentencepiece models support
26
- sentencepiece==0.1.99
 
1
+ - gradio==4.44.0
2
+ - gradio_client==0.17.0
3
+ - # ... (other stuff)
4
+ +# --- UI stack (pin these) ---
5
+ +gradio==4.44.0
6
+ +# gradio-client 1.x is pulled automatically by gradio 4.44.0; don't pin an old one
7
+ +# Ensure the modern asyncio package path exists:
8
+ +websockets>=14,<16
9
+ +
10
+ +# --- If you actually need these, keep/pin them; otherwise remove to speed up builds ---
11
+ +fastapi==0.115.0
12
+ +uvicorn==0.30.6
13
+ +pydantic<3
14
+ +
15
+ +# --- Your model stack (only if used) ---
16
+ +# transformers==4.44.2
17
+ +# accelerate>=0.34.0
18
+ +# bitsandbytes==0.43.1 # Only if you really need bnb; not for CPU-only ZeroGPU runs