ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-13 06:29:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed issue for vaes without a shift

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-11 10:30:55 -06:00
parent 8d48ad4e85
commit 6490a326e5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/stable_diffusion_model.py
View File

@@ -1960,7 +1960,8 @@ class StableDiffusion:
else:
latents = self.vae.encode(images).latent_dist.sample()
# latents = self.vae.encode(images, return_dict=False)[0]
latents = latents * (self.vae.config['scaling_factor'] - self.vae.config['shift_factor'])
shift = self.vae.config['shift_factor'] if self.vae.config['shift_factor'] is not None else 0
latents = latents * (self.vae.config['scaling_factor'] - shift)
latents = latents.to(device, dtype=dtype)
return latents