drewwas commited on
Commit
49e57e0
1 Parent(s): 140aee9

Create requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -0
requirements.txt ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ transformers==4.45.2
2
+ datasets==2.19.2
3
+ transformer-tricks>=0.2.3
4
+ accelerate>=0.9.0
5
+ safetensors>=0.4.5
6
+ huggingface-hub>=0.25.1
7
+ torch>=2.4.1
8
+ tqdm>=4.64.1
9
+ jupytext>=1.16.4
10
+ autopep8>=2.3.1
11
+ twine>=5.1.1
12
+ build>=1.2.2
13
+
14
+ # pip list # see all versions
15
+ #
16
+ # Phi-3 needs flash-attn, but this requires CUDA
17
+ # flash-attn==2.5.8
18
+