ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-06-28 05:58:49 +00:00
commit f408bcb6cd
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/backend/core/inference/diffusion_attention.py
View file

@ -71,8 +71,8 @@ def normalize_attention_backend(value: Optional[str]) -> Optional[str]:
# and then crashes mid-generation. Gate them up front: minimum (major, minor) compute
# capability per dispatcher backend name.
_MIN_CUDA_CAPABILITY: dict[str, tuple[int, int]] = {
"_flash_3_hub": (9, 0), # FlashAttention 3 -> Hopper (SM90)
"flash_4_hub": (10, 0), # FlashAttention 4 -> Blackwell (SM100)
"_flash_3_hub": (9, 0), # FlashAttention 3 -> Hopper (SM90)
"flash_4_hub": (10, 0), # FlashAttention 4 -> Blackwell (SM100)
}