Wenye He commited on
Commit
2c277a8
·
verified ·
1 Parent(s): c5b2064

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
  torch==2.1.2+cu121
3
- flash-attn==2.7.3.post1
4
  transformers==4.42.0
5
  gradio==4.29.0
6
  accelerate==0.30.0
 
1
  --extra-index-url https://download.pytorch.org/whl/cu121
2
  torch==2.1.2+cu121
3
+ flash-attn
4
  transformers==4.42.0
5
  gradio==4.29.0
6
  accelerate==0.30.0