Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
@@ -4,4 +4,6 @@ transformers
|
|
4 |
gradio
|
5 |
pillow
|
6 |
einops
|
7 |
-
timm
|
|
|
|
|
|
4 |
gradio
|
5 |
pillow
|
6 |
einops
|
7 |
+
timm
|
8 |
+
accelerate>=0.26.0
|
9 |
+
flash-attention
|