Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix vae decode for backend original

Browse Source
This commit is contained in:
Vladimir Mandic
2024-09-02 08:23:40 -04:00
parent 7f2e30576d
commit 045b24c060
5 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_helpers.py
+2 -1
View File
@@ -196,7 +196,8 @@ def decode_first_stage(model, x, full_quality=True):
try:
if full_quality:
if hasattr(model, 'decode_first_stage'):
x_sample = model.decode_first_stage(x) * 0.5 + 0.5
# x_sample = model.decode_first_stage(x) * 0.5 + 0.5
x_sample = model.decode_first_stage(x)
elif hasattr(model, 'vae'):
x_sample = processing_vae.vae_decode(latents=x, model=model, output_type='np', full_quality=full_quality)
else: