Update requirements.txt
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -2,6 +2,7 @@ pillow
|
|
| 2 |
einops
|
| 3 |
spaces
|
| 4 |
torch
|
|
|
|
| 5 |
timm
|
| 6 |
transformers==4.42.0
|
| 7 |
supervision==0.25.1
|
|
|
|
| 2 |
einops
|
| 3 |
spaces
|
| 4 |
torch
|
| 5 |
+
flash_attn
|
| 6 |
timm
|
| 7 |
transformers==4.42.0
|
| 8 |
supervision==0.25.1
|