Update requirements.txt
Browse files- requirements.txt +3 -1
requirements.txt
CHANGED
@@ -2,4 +2,6 @@ accelerate @ git+https://github.com/huggingface/accelerate
|
|
2 |
transformers
|
3 |
timm
|
4 |
huggingface_hub
|
5 |
-
tabulate
|
|
|
|
|
|
2 |
transformers
|
3 |
timm
|
4 |
huggingface_hub
|
5 |
+
tabulate
|
6 |
+
flash-attn
|
7 |
+
einops
|