Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add shift_factor to vae decode

Browse Source
This commit is contained in:
Disty0
2024-07-23 02:19:30 +03:00
parent f2769c0449
commit 4aabc8b614
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_vae.py
+3
View File
@@ -51,12 +51,15 @@ def full_vae_decode(latents, model):
latents_mean = model.vae.config.get("latents_mean", None)
latents_std = model.vae.config.get("latents_std", None)
scaling_factor = model.vae.config.get("scaling_factor", None)
shift_factor = model.vae.config.get("shift_factor", None)
if latents_mean and latents_std:
latents_mean = (torch.tensor(latents_mean).view(1, 4, 1, 1).to(latents.device, latents.dtype))
latents_std = (torch.tensor(latents_std).view(1, 4, 1, 1).to(latents.device, latents.dtype))
latents = latents * latents_std / scaling_factor + latents_mean
else:
latents = latents / scaling_factor
if shift_factor:
latents = latents + shift_factor
decoded = model.vae.decode(latents, return_dict=False)[0]
# delete vae after OpenVINO compile