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-02 04:38:21 +00:00
commit a3e23a8080
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
studio/backend/tests/test_diffusion_sdxl.py
View file

@ -138,7 +138,7 @@ def test_align_vae_dtype_skips_gguf_packed_uint8_params():
class _GgufDenoiser:
def parameters(self):
yield types.SimpleNamespace(dtype = torch.uint8) # packed GGUF block
yield types.SimpleNamespace(dtype = torch.uint8) # packed GGUF block
yield types.SimpleNamespace(dtype = torch.bfloat16) # compute dtype
vae = _FakeVae(dtype = torch.float32)
@ -151,9 +151,7 @@ def test_align_vae_dtype_skips_gguf_packed_uint8_params():
yield types.SimpleNamespace(dtype = torch.uint8)
vae2 = _FakeVae(dtype = torch.float32)
DiffusionBackend._align_vae_dtype(
types.SimpleNamespace(transformer = _AllPacked(), vae = vae2)
)
DiffusionBackend._align_vae_dtype(types.SimpleNamespace(transformer = _AllPacked(), vae = vae2))
assert vae2.moved_to is None