CamiloVega commited on
Commit
8573f3c
·
verified ·
1 Parent(s): 2c614f8

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +23 -1
requirements.txt CHANGED
@@ -1,6 +1,28 @@
 
1
  transformers==4.36.2
2
  torch==2.4.0
3
  accelerate==0.27.2
4
  gradio==4.19.2
5
  huggingface-hub==0.20.3
6
- bitsandbytes==0.41.3
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core dependencies
2
  transformers==4.36.2
3
  torch==2.4.0
4
  accelerate==0.27.2
5
  gradio==4.19.2
6
  huggingface-hub==0.20.3
7
+
8
+ # Model optimization and quantization
9
+ bitsandbytes==0.41.3
10
+ scipy==1.11.4
11
+ einops==0.7.0
12
+ sentencepiece==0.1.99
13
+
14
+ # Data processing and utilities
15
+ numpy==1.24.3
16
+ pandas==2.1.4
17
+ tqdm==4.66.1
18
+
19
+ # Memory management
20
+ psutil==5.9.7
21
+
22
+ # Optional but recommended for better performance
23
+ datasets==2.15.0
24
+ tokenizers==0.15.0
25
+ safetensors==0.4.1
26
+
27
+ # Logging and monitoring
28
+ tensorboard==2.15.1