Hugging Face's logo

Spaces:
BAAI
/
tokenize-anything
Running on A10G

App Files Community
1
PhyscalX commited on
Commit
4fa3572
•
1 Parent(s): 3d2142b

Fix FlashAttn installation

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  opencv-python
2
  Pillow
3
  gradio-image-prompter
4
- torch>=2.0.0
5
- flash-attn>=2.3.3
 
1
  opencv-python
2
  Pillow
3
  gradio-image-prompter
4
+ torch==2.1.0
5
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.6/flash_attn-2.3.6+cu122torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl