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-07-06 14:05:20 +00:00
commit d86c4c70e6
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
studio/backend/tests/test_diffusion_backend.py
View file

@ -2134,11 +2134,15 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch):
# memory_mode overrides it) must NOT widen, even though the candidate itself is dense-viable.
before = len(seen)
assert (
backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "balanced"})
backend._dense_quant_prefetch_needed(
fam, {"transformer_quant": "fp8", "memory_mode": "balanced"}
)
is False
)
assert (
backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "low_vram"})
backend._dense_quant_prefetch_needed(
fam, {"transformer_quant": "fp8", "memory_mode": "low_vram"}
)
is False
)
assert (
@ -2150,7 +2154,9 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch):
# An explicit memory_mode still consulting the candidate: fast/auto can flip resident, so they
# widen when the candidate is dense-viable (memory_mode="fast" does not force offload).
assert (
backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "fast"})
backend._dense_quant_prefetch_needed(
fam, {"transformer_quant": "fp8", "memory_mode": "fast"}
)
is True
)
# A cpu_offload flag is overridden by an explicit resident memory_mode, so it still widens.