Ketengan-Diffusion-Lab commited on
Commit
f3c7bef
1 Parent(s): fca9b9b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,5 +1,7 @@
1
  torchvision
2
  torch
 
 
3
  transformers
4
  gradio
5
  spaces
 
1
  torchvision
2
  torch
3
+ ninja
4
+ git+https://github.com/Dao-AILab/flash-attention.git
5
  transformers
6
  gradio
7
  spaces