ColdSlim commited on
Commit
fd2ac66
·
verified ·
1 Parent(s): c8b3c1b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +15 -11
requirements.txt CHANGED
@@ -1,12 +1,16 @@
1
- # requirements.txt (append or replace the transformers line)
2
- git+https://github.com/huggingface/transformers.git
3
- peft>=0.11.0
4
- accelerate
5
- gradio>=4.37.2 # any 4.37+ or 5.x is fine
6
- gradio_client>=0.20.0
7
- huggingface_hub>=0.30.0,<1.0
8
- pillow
 
 
 
9
 
10
- # --- core ---
11
- torch==2.1.2 # pin to avoid pip pulling 2.7.x
12
- torchvision==0.16.2 # matching wheel for 2.1.x
 
 
1
+ # --- core (compatible with ZeroGPU supported stack) ---
2
+ numpy<2.0
3
+ torch==2.1.2
4
+ torchvision==0.16.2
5
+ safetensors>=0.4.3
6
+
7
+ # --- HF stack (pin; avoid git master) ---
8
+ transformers==4.43.3
9
+ accelerate==0.31.0
10
+ peft==0.11.1
11
+ huggingface_hub>=0.23,<1.0
12
 
13
+ # --- UI / misc ---
14
+ gradio>=4.37.2,<5.1
15
+ gradio_client>=0.20.0
16
+ pillow