From d756489a5c8ed9c74fdfc51b1160ea1bf1675861 Mon Sep 17 00:00:00 2001 From: Victor Giers Date: Sun, 30 Nov 2025 23:27:10 +0100 Subject: [PATCH] Adjust device context and add generator for non-CPU devices --- concept-maker_gui.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/concept-maker_gui.py b/concept-maker_gui.py index 183034d..26f2b9e 100644 --- a/concept-maker_gui.py +++ b/concept-maker_gui.py @@ -1953,9 +1953,10 @@ class App(TkinterDnD.Tk): # type: ignore output_dir.mkdir(parents=True, exist_ok=True) ctx = ( torch.autocast(device_type=device, dtype=torch.float16) - if device in {"cuda", "mps"} + if device == "cuda" else contextlib.nullcontext() ) + generator = torch.Generator(device=device) if device != "cpu" else None with torch.inference_mode(): with ctx: res = pipe( @@ -1963,6 +1964,9 @@ class App(TkinterDnD.Tk): # type: ignore guidance_scale=2.0, num_inference_steps=8, num_images_per_prompt=1, + height=1024, + width=1024, + generator=generator, ) img = res.images[0] slug = self._slug(self.title_var.get().strip() or "image")