Alexandre-Numind commited on
Commit
6fa892e
·
verified ·
1 Parent(s): f9933eb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- transformers
2
- torch
3
- accelerate
4
- flash_attn
 
1
+ transformers==4.52.3
2
+ torch==2.6.0
3
+ accelerate==0.32.0
4
+ flash-attn==2.7.3