rajeshthangaraj1 commited on
Commit
856cca8
·
verified ·
1 Parent(s): db95b22

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,5 +1,5 @@
 
1
  torch
2
- flash_attn
3
  transformers
4
  accelerate
5
  bitsandbytes
 
1
+ flash-attn==2.6.2
2
  torch
 
3
  transformers
4
  accelerate
5
  bitsandbytes