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 08:49:27 +00:00
commit 7c72910487
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2132,9 +2132,7 @@ def test_plan_memory_dense_replan_does_not_double_count_prefetched_transformer(m
from core.inference.diffusion_memory import OFFLOAD_NONE, DeviceMemory
backend = DiffusionBackend()
target = types.SimpleNamespace(
device = "cuda", backend = "cuda", supports_model_cpu_offload = True
)
target = types.SimpleNamespace(device = "cuda", backend = "cuda", supports_model_cpu_offload = True)
# 40 GiB discrete card, 40000 MiB free: comfortably fits transformer + real
# companions + headroom, but NOT a second copy of the bf16 transformer.
monkeypatch.setattr(