CooLLaMACEO commited on
Commit
0196236
·
verified ·
1 Parent(s): 386eb5c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +15 -8
requirements.txt CHANGED
@@ -1,17 +1,24 @@
1
- # Use the CPU-optimized version of Torch to save space
 
 
 
2
  --extra-index-url https://download.pytorch.org/whl/cpu
3
  torch==2.2.1
4
  torchvision==0.17.1
5
  torchaudio==2.2.1
6
 
7
- # Essential for the 1-bit logic and model loading
8
- transformers>=4.38.0
 
 
 
 
 
9
  accelerate>=0.27.0
10
  safetensors>=0.4.0
11
-
12
- # Interface and API
13
- gradio>=4.0.0
14
  huggingface_hub>=0.20.0
15
 
16
- # Optional: optimized memory management
17
- psutil
 
 
 
1
+ # Force NumPy 1.x to prevent the Torch crash
2
+ numpy<2.0.0
3
+
4
+ # CPU-optimized Torch
5
  --extra-index-url https://download.pytorch.org/whl/cpu
6
  torch==2.2.1
7
  torchvision==0.17.1
8
  torchaudio==2.2.1
9
 
10
+ # Missing Tokenizer Dependencies
11
+ protobuf
12
+ sentencepiece
13
+ tiktoken
14
+
15
+ # Core Logic
16
+ transformers==4.38.0
17
  accelerate>=0.27.0
18
  safetensors>=0.4.0
 
 
 
19
  huggingface_hub>=0.20.0
20
 
21
+ # API Server
22
+ fastapi
23
+ uvicorn
24
+ pydantic