Hugging Face's logo

Spaces:
prithivMLmods
/
Tiny-VLMs-Lab
Running on Zero

App Files Community
12
prithivMLmods commited on
Commit
b537cee
·
verified ·
1 Parent(s): 52160be

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  torch
2
  torchvision
3
- https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.8/flash_attn-2.7.4.post1+cu126torch2.7-cp310-cp310-linux_x86_64.whl
4
  transformers==4.51.3
5
  transformers-stream-generator
6
  qwen-vl-utils
 
1
  torch
2
  torchvision
3
+ flash-attn==2.8.0.post2
4
  transformers==4.51.3
5
  transformers-stream-generator
6
  qwen-vl-utils