Benjamin Gonzalez commited on
Commit
f647657
1 Parent(s): 38c7933

add rest of requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +21 -1
requirements.txt CHANGED
@@ -1,2 +1,22 @@
1
- transformers==4.36.0
2
  flash-attn==2.3.6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  flash-attn==2.3.6
2
+ mlflow==2.6.0
3
+ cloudpickle==2.2.1
4
+ jsonpickle==3.0.1
5
+ mlflow-skinny==2.6.0
6
+ azureml-core==1.51.0.post1
7
+ azureml-mlflow==1.51.0
8
+ azureml-metrics[all]==0.0.32
9
+ scikit-learn==1.2.2
10
+ cryptography==41.0.1
11
+ python-dateutil==2.8.2
12
+ datasets==2.14.6
13
+ soundfile==0.12.1
14
+ librosa==0.10.1
15
+ diffusers==0.21.4
16
+ sentencepiece==0.1.99
17
+ transformers==4.34.0
18
+ torch==2.1.0
19
+ accelerate==0.23.0
20
+ Pillow==9.4.0
21
+ einops
22
+ azureml-evaluate-mlflow==0.0.32