Update requirements.txt
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -7,4 +7,5 @@ einops
|
|
| 7 |
omegaconf
|
| 8 |
safetensors
|
| 9 |
torch==2.5.1
|
| 10 |
-
torchvision==0.20.1
|
|
|
|
|
|
| 7 |
omegaconf
|
| 8 |
safetensors
|
| 9 |
torch==2.5.1
|
| 10 |
+
torchvision==0.20.1
|
| 11 |
+
flash-attn==2.6.3
|