Merge branch 'image-generation' of https://github.com/unslothai/unsloth into image-generation

studio/backend/core/inference/diffusion.py

@@ -1086,9 +1086,7 @@ class DiffusionBackend:
                         pipe_kwargs = {"torch_dtype": dtype, "transformer": transformer}
                         if hf_token:
                             pipe_kwargs["token"] = hf_token
-                        pipe = pipeline_cls.from_pretrained(
-                            _base_local_dir or base, **pipe_kwargs
-                        )
+                        pipe = pipeline_cls.from_pretrained(_base_local_dir or base, **pipe_kwargs)
 
                 # Resolve the effective speed mode: GGUF models default to the
                 # near-lossless `default` profile (compile is ~2.2x and sits below

studio/backend/tests/test_diffusion_backend.py

@@ -2136,8 +2136,7 @@ def test_prefetch_returns_snapshot_dir_for_manifest(monkeypatch):
     )
     assert root == "/cache/snap"
     assert (
-        backend._prefetch_files("base/repo", None, "base/repo", ["vae/x.safetensors"], None)
-        is None
+        backend._prefetch_files("base/repo", None, "base/repo", ["vae/x.safetensors"], None) is None
     )