From e6bd256ad6789724461456ca568441fb9b7e1be6 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 6 Jul 2026 10:50:41 +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.py | 4 +++- studio/backend/tests/test_diffusion_backend.py | 14 ++++++++++---- 2 files changed, 13 insertions(+), 5 deletions(-) diff --git a/studio/backend/core/inference/diffusion.py b/studio/backend/core/inference/diffusion.py index 767961695d..1ae3ce2668 100644 --- a/studio/backend/core/inference/diffusion.py +++ b/studio/backend/core/inference/diffusion.py @@ -1030,7 +1030,9 @@ class DiffusionBackend: # break the user's bit-exact request (an auto DEFAULT overriding an EXPLICIT # control). Suppress the auto default when speed was explicitly pinned off; # otherwise auto (the dense-capable default) applies. - speed_off = speed_mode is not None and str(speed_mode).strip().lower() == SPEED_OFF + speed_off = ( + speed_mode is not None and str(speed_mode).strip().lower() == SPEED_OFF + ) # "off" normalizes to None (no dense quant), keeping the GGUF-as-is path. transformer_quant = "off" if speed_off else TQ_AUTO diff --git a/studio/backend/tests/test_diffusion_backend.py b/studio/backend/tests/test_diffusion_backend.py index 155daff8b1..f210c73b9e 100644 --- a/studio/backend/tests/test_diffusion_backend.py +++ b/studio/backend/tests/test_diffusion_backend.py @@ -2105,7 +2105,15 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch): seen: list = [] - def fake_candidate(*, fam, target, requested, base_repo = None, prequant_path = None, logger = None): + def fake_candidate( + *, + fam, + target, + requested, + base_repo = None, + prequant_path = None, + logger = None, + ): seen.append(requested) return object() # a viable dense-quant candidate (scheme resolves AND disk fits) @@ -2121,9 +2129,7 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch): assert backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "none"}) is False # An explicit Speed="off" (bit-exact) load suppresses the dense path -> never widen. assert ( - backend._dense_quant_prefetch_needed( - fam, {"transformer_quant": "fp8", "speed_mode": "off"} - ) + backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "speed_mode": "off"}) is False ) # No viable candidate (unsupported scheme / no disk room / a prequant checkpoint shortcut)