Fix variable name

This commit is contained in:
ZachNagengast 2023-07-27 23:40:23 -07:00
parent aa1f827271
commit e6d14c708c

View File

@ -548,7 +548,7 @@ class SDXLLatentsToLatentsInvocation(BaseInvocation):
with unet_info as unet:
# apply denoising_start
num_inference_steps = self.steps
scheduler.set_timesteps(num_inference_steps, device=self.scheduler.device)
scheduler.set_timesteps(num_inference_steps, device=unet.device)
t_start = int(round(self.denoising_start * num_inference_steps))
timesteps = scheduler.timesteps[t_start * scheduler.order :]