Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-11-04 12:35:38 -05:00
parent a2f9a4dbb0
commit bb80286aab
+1
View File
@@ -289,6 +289,7 @@ class StableDiffusionXLPuLIDPipeline:
"""
raise NotImplementedError('pulid: img2img')
else:
# standard txt2img will full noise
latents = torch.randn((size[0], 4, size[1] // 8, size[2] // 8), device="cpu", generator=torch.manual_seed(seed))
latents = latents.to(dtype=self.pipe.unet.dtype, device=self.device)
return latents