Tonic commited on
Commit
0950b55
1 Parent(s): e042085

add dependencies

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -1
requirements.txt CHANGED
@@ -1 +1,6 @@
1
- huggingface_hub==0.22.2
 
 
 
 
 
 
1
+ huggingface_hub==0.22.2
2
+ torch==2.2.0
3
+ transformers
4
+ flash-attn==2.2.0
5
+ einops
6
+ sentence-transformers==2.7.0