CooLLaMACEO commited on
Commit
33f722d
·
verified ·
1 Parent(s): 2353df0

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -0
requirements.txt ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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