Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ flash_attention_installed = False
|
|
| 21 |
try:
|
| 22 |
print("Attempting to download and install FlashAttention wheel...")
|
| 23 |
flash_attention_wheel = hf_hub_download(
|
| 24 |
-
repo_id="
|
| 25 |
repo_type="model",
|
| 26 |
filename="128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl",
|
| 27 |
)
|
|
|
|
| 21 |
try:
|
| 22 |
print("Attempting to download and install FlashAttention wheel...")
|
| 23 |
flash_attention_wheel = hf_hub_download(
|
| 24 |
+
repo_id="rahul7star/flash-attn-3",
|
| 25 |
repo_type="model",
|
| 26 |
filename="128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl",
|
| 27 |
)
|