From e73444ed992699129394bd84ff9a9db8861ecd61 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 4 Jul 2026 07:47:47 +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_speed.py | 6 +++++- studio/backend/routes/inference.py | 1 - studio/backend/tests/test_diffusion_speed.py | 11 +++++++---- 3 files changed, 12 insertions(+), 6 deletions(-) diff --git a/studio/backend/core/inference/diffusion_speed.py b/studio/backend/core/inference/diffusion_speed.py index dd2f2e664d..cb9101d03d 100644 --- a/studio/backend/core/inference/diffusion_speed.py +++ b/studio/backend/core/inference/diffusion_speed.py @@ -357,7 +357,11 @@ _FP16_ACCUM_DENY: frozenset[str] = frozenset() def _enable_fp16_accumulation( - family: Any, logger: Any, *, dtype: Any = None, speed_mode: Optional[str] = None + family: Any, + logger: Any, + *, + dtype: Any = None, + speed_mode: Optional[str] = None, ) -> bool: """Turn on fp16-accumulated fp16 GEMMs for consumer GPUs, where they run ~2x the fp32-accumulate rate (datacenter HBM parts are not throughput-nerfed, so they keep diff --git a/studio/backend/routes/inference.py b/studio/backend/routes/inference.py index 23ae5c7a6e..2c1a4f5c3f 100644 --- a/studio/backend/routes/inference.py +++ b/studio/backend/routes/inference.py @@ -11980,7 +11980,6 @@ async def diffusion_inference_info(current_subject: str = Depends(get_current_su Hardware-independent (served from the pure auto-policy tables, no GPU probing), so it is cheap and safe to fetch before anything is loaded.""" from core.inference.diffusion_inference_info import family_inference_infos - return DiffusionInferenceInfoResponse(families = family_inference_infos()) diff --git a/studio/backend/tests/test_diffusion_speed.py b/studio/backend/tests/test_diffusion_speed.py index 9dc50f4e28..5b645022e1 100644 --- a/studio/backend/tests/test_diffusion_speed.py +++ b/studio/backend/tests/test_diffusion_speed.py @@ -357,7 +357,12 @@ def test_apply_tolerates_missing_optims(monkeypatch): # ── fp16 accumulation (consumer fp16-GEMM fast path) ────────────────────────── -def _stub_torch_fp16_accum(monkeypatch, *, consumer = True, with_flag = True): +def _stub_torch_fp16_accum( + monkeypatch, + *, + consumer = True, + with_flag = True, +): torch = types.ModuleType("torch") torch.bfloat16 = "bfloat16" torch.channels_last = "channels_last" @@ -427,9 +432,7 @@ def test_fp16_accum_respects_family_deny_list(monkeypatch): _stub_gguf_accel(monkeypatch) monkeypatch.setattr(ds_mod, "_FP16_ACCUM_DENY", frozenset({"fragile-family"})) fam = types.SimpleNamespace(supports_torch_compile = True, name = "fragile-family") - applied = apply_speed_optims( - _Pipe(), _target(), is_gguf = True, family = fam, speed_mode = "default" - ) + applied = apply_speed_optims(_Pipe(), _target(), is_gguf = True, family = fam, speed_mode = "default") assert applied["fp16_accum"] is False