Hugging Face's logo

Spaces:
Asahina2K
/
danbooru-tags-transformer
Runtime error

App Files Community
p1atdev commited on
Commit
400e678
•
1 Parent(s): 9e7ee0b

chore: add flash attn

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  torch==2.1.0
2
  transformers==4.38.0
3
- optimum[onnxruntime]==1.17.1
 
 
1
  torch==2.1.0
2
  transformers==4.38.0
3
+ optimum[onnxruntime]==1.17.1
4
+ flash_attn