Spaces:
Running
on
Zero
Running
on
Zero
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
|