ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-27 17:51:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added this not that guidance. Added ability to replace prompts.

Browse Source
This commit is contained in:
Jaret Burkett
2024-02-28 20:10:14 -07:00
parent 561914d8e6
commit 337945de9a
7 changed files with 114 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
toolkit/stable_diffusion_model.py
View File

@@ -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)