From e342ca872f33a70097f615dbd6d3b92df5851679 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 28 Jul 2023 10:27:49 -0400 Subject: [PATCH] fix to work on non-MPS systems --- invokeai/app/invocations/sdxl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invokeai/app/invocations/sdxl.py b/invokeai/app/invocations/sdxl.py index 772f422c0f..0016960b37 100644 --- a/invokeai/app/invocations/sdxl.py +++ b/invokeai/app/invocations/sdxl.py @@ -293,14 +293,14 @@ class SDXLTextToLatentsInvocation(BaseInvocation): num_inference_steps = self.steps - latents = latents * scheduler.init_noise_sigma - unet_info = context.services.model_manager.get_model(**self.unet.unet.dict(), context=context) do_classifier_free_guidance = True cross_attention_kwargs = None with unet_info as unet: scheduler.set_timesteps(num_inference_steps, device=unet.device) timesteps = scheduler.timesteps + + latents = latents.to(device=unet.device, dtype=unet.dtype) * scheduler.init_noise_sigma extra_step_kwargs = dict() if "eta" in set(inspect.signature(scheduler.step).parameters.keys()):