Vikaspandey582003 commited on
Commit
093e7af
·
verified ·
1 Parent(s): fc58aef

fix: remove unsloth/trl/bitsandbytes from Space requirements (training only, use Colab)

Browse files
Files changed (1) hide show
  1. requirements.txt +21 -23
requirements.txt CHANGED
@@ -1,42 +1,40 @@
1
- # Core ML
 
 
 
 
 
 
 
 
 
2
  torch>=2.1.0
3
  transformers>=4.44.0
4
- trl>=0.9.0
5
  datasets>=2.18.0
6
  accelerate>=0.28.0
7
  peft>=0.10.0
8
- bitsandbytes>=0.42.0
9
-
10
- # Unsloth — 2-3x faster training, 70% less VRAM (install first on GPU machines)
11
- unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
12
-
13
- # Optional: GPT-4o-mini baseline comparison
14
- openai>=1.0.0
15
 
16
- # Environment
17
  gymnasium>=1.0.0
18
  numpy>=1.26.0
19
  pandas>=2.1.0
20
  scipy>=1.11.0
21
-
22
- # Server
23
- fastapi>=0.111.0
24
- uvicorn[standard]>=0.29.0
25
- pydantic>=2.6.0
26
- httpx>=0.27.0
27
-
28
- # Demo
29
- gradio>=4.20.0
30
 
31
  # Visualization
32
  matplotlib>=3.8.0
33
  seaborn>=0.13.0
 
34
 
35
  # Utilities
36
- wandb>=0.16.0
37
  huggingface-hub>=0.21.0
38
- scikit-learn>=1.4.0
39
  python-dotenv>=1.0.0
40
- click>=8.1.0
41
- rich>=13.0.0
42
  PyYAML>=6.0.0
 
 
 
 
 
 
 
 
 
1
+ # Server (required for Space)
2
+ fastapi>=0.111.0
3
+ uvicorn[standard]>=0.29.0
4
+ pydantic>=2.6.0
5
+ httpx>=0.27.0
6
+
7
+ # UI
8
+ gradio>=4.20.0
9
+
10
+ # ML inference (lightweight — no training on Space)
11
  torch>=2.1.0
12
  transformers>=4.44.0
 
13
  datasets>=2.18.0
14
  accelerate>=0.28.0
15
  peft>=0.10.0
 
 
 
 
 
 
 
16
 
17
+ # Environment / calibration
18
  gymnasium>=1.0.0
19
  numpy>=1.26.0
20
  pandas>=2.1.0
21
  scipy>=1.11.0
22
+ scikit-learn>=1.4.0
 
 
 
 
 
 
 
 
23
 
24
  # Visualization
25
  matplotlib>=3.8.0
26
  seaborn>=0.13.0
27
+ plotly>=5.0.0
28
 
29
  # Utilities
 
30
  huggingface-hub>=0.21.0
 
31
  python-dotenv>=1.0.0
 
 
32
  PyYAML>=6.0.0
33
+ rich>=13.0.0
34
+
35
+ # Optional — only needed for training (Colab/GPU), not for Space server
36
+ # trl>=0.9.0
37
+ # bitsandbytes>=0.42.0
38
+ # unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
39
+ # openai>=1.0.0
40
+ # wandb>=0.16.0