From f408bcb6cd8d4f012ad9050d61c89d0cdba3b8dc Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 28 Jun 2026 05:58:49 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/inference/diffusion_attention.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/studio/backend/core/inference/diffusion_attention.py b/studio/backend/core/inference/diffusion_attention.py index eea5e91623..0dc007a9f1 100644 --- a/studio/backend/core/inference/diffusion_attention.py +++ b/studio/backend/core/inference/diffusion_attention.py @@ -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) }