Added this not that guidance. Added ability to replace prompts.
This commit is contained in:
1 parent
561914d8e6
commit
337945de9a
7 files changed
+114
-5
No files matched your search
@@ -1273,6 +1273,7 @@ class StableDiffusion:
|
||||
|
||||
images = torch.stack(image_list)
|
||||
latents = self.vae.encode(images).latent_dist.sample()
|
||||
# latents = self.vae.encode(images, return_dict=False)[0]
|
||||
latents = latents * self.vae.config['scaling_factor']
|
||||
latents = latents.to(device, dtype=dtype)
|
||||
|
||||
|
||||
Reference in new issue
Block a user