Skip to content

Commit 25c701d

Browse files
committed
up
1 parent a1e1faf commit 25c701d

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/diffusers/models/attention_dispatch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -369,7 +369,7 @@ def _check_attention_backend_requirements(backend: AttentionBackendName) -> None
369369
elif backend in [AttentionBackendName._FLASH_3_HUB]:
370370
if not DIFFUSERS_ENABLE_HUB_KERNELS:
371371
raise RuntimeError(
372-
f"Flash Attention 3 Hub backend '{backend.value}' is not usable because the `DIFFUSERS_ENABLE_HUB_KERNELS` env var isn't set. Please set it `DIFFUSERS_ENABLE_HUB_KERNELS=yes`."
372+
f"Flash Attention 3 Hub backend '{backend.value}' is not usable because the `DIFFUSERS_ENABLE_HUB_KERNELS` env var isn't set. Please set it like `export DIFFUSERS_ENABLE_HUB_KERNELS=yes`."
373373
)
374374
if not is_kernels_available():
375375
raise RuntimeError(

src/diffusers/utils/kernels_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def _get_fa3_from_hub():
1616

1717
try:
1818
# TODO: temporary revision for now. Remove when merged upstream into `main`.
19-
flash_attn_3_hub = get_kernel(_DEFAULT_HUB_ID_FA3, revision="fake-ops-return-prob")
19+
flash_attn_3_hub = get_kernel(_DEFAULT_HUB_ID_FA3, revision="fake-ops-return-probs")
2020
return flash_attn_3_hub
2121
except Exception as e:
2222
logger.error(f"An error occurred while fetching kernel '{_DEFAULT_HUB_ID_FA3}' from the Hub: {e}")

0 commit comments

Comments
 (0)