sounar commited on
Commit
26d8378
·
verified ·
1 Parent(s): 20d06bb

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  torch==2.5.1
2
  transformers==4.46.3
3
  bitsandbytes==0.44.1
@@ -5,4 +6,4 @@ Pillow==11.0.0
5
  torchvision==0.20.1
6
  accelerate==1.1.1
7
  flask==3.1.0
8
- flash-attn==2.7.0.post2
 
1
+ wheel==0.45.0
2
  torch==2.5.1
3
  transformers==4.46.3
4
  bitsandbytes==0.44.1
 
6
  torchvision==0.20.1
7
  accelerate==1.1.1
8
  flask==3.1.0
9
+ flash-attn==2.7.0.post2