teowu commited on
Commit
ba33cb5
1 Parent(s): 56c0224

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  icecream
2
  markdown2
3
- flash_attn
4
  pydantic
5
  accelerate==0.21.0
6
  transformers==4.36.2
@@ -17,4 +17,5 @@ einops==0.6.1
17
  scikit-learn==1.2.2
18
  numpy
19
  gradio==3.50.0
20
- gradio_client==0.6.1
 
 
1
  icecream
2
  markdown2
3
+
4
  pydantic
5
  accelerate==0.21.0
6
  transformers==4.36.2
 
17
  scikit-learn==1.2.2
18
  numpy
19
  gradio==3.50.0
20
+ gradio_client==0.6.1
21
+ flash_attn