Spaces:
Running
on
Zero
Running
on
Zero
yuhangzang
commited on
Commit
·
d9d4ac7
1
Parent(s):
fd68401
update
Browse files- requirements.txt +2 -1
requirements.txt
CHANGED
|
@@ -5,6 +5,7 @@ gradio>=5.49.1
|
|
| 5 |
spaces>=0.24.0
|
| 6 |
pillow
|
| 7 |
torchvision
|
| 8 |
-
|
|
|
|
| 9 |
# This installs only on 64-bit Linux. It will be skipped on macOS/Windows/ARM.
|
| 10 |
flash-attn; platform_system == "Linux" and platform_machine == "x86_64"
|
|
|
|
| 5 |
spaces>=0.24.0
|
| 6 |
pillow
|
| 7 |
torchvision
|
| 8 |
+
|
| 9 |
+
# Optional: FlashAttention v2 for faster attention on compatible Linux CUDA GPUs.
|
| 10 |
# This installs only on 64-bit Linux. It will be skipped on macOS/Windows/ARM.
|
| 11 |
flash-attn; platform_system == "Linux" and platform_machine == "x86_64"
|