likhonsheikh commited on
Commit
63a6933
·
verified ·
1 Parent(s): d4cfbf8

Add requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +44 -0
requirements.txt ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Core Dependencies
2
+ torch>=2.0.0
3
+ transformers>=4.37.0
4
+ tokenizers>=0.15.0
5
+ datasets>=2.14.0
6
+ accelerate>=0.24.0
7
+
8
+ # Training Dependencies
9
+ deepspeed>=0.12.0
10
+ flash-attn>=2.3.0
11
+ xformers>=0.0.22
12
+
13
+ # Evaluation Dependencies
14
+ evaluate>=0.4.0
15
+ huggingface-hub>=0.17.0
16
+ numpy>=1.24.0
17
+ scipy>=1.10.0
18
+ scikit-learn>=1.3.0
19
+
20
+ # Monitoring and Logging
21
+ wandb>=0.15.0
22
+ tensorboard>=2.13.0
23
+ mlflow>=2.7.0
24
+
25
+ # Code Quality and Testing
26
+ pytest>=7.4.0
27
+ black>=23.0.0
28
+ isort>=5.12.0
29
+ flake8>=6.0.0
30
+
31
+ # Quantization
32
+ bitsandbytes>=0.41.0
33
+ triton>=2.0.0
34
+
35
+ # Development
36
+ jupyter>=1.0.0
37
+ ipywidgets>=8.0.0
38
+ matplotlib>=3.7.0
39
+ seaborn>=0.12.0
40
+
41
+ # Optional Dependencies for Mobile
42
+ onnx>=1.14.0
43
+ onnxruntime>=1.15.0
44
+ optimum>=1.11.0