Fix Cascade and add full_vqgan_decode

This commit is contained in:
Disty0
2024-11-28 22:12:25 +03:00
parent 425f51bb25
commit 75dd6219c6
3 changed files with 65 additions and 3 deletions
+1 -1
View File
@@ -352,7 +352,7 @@ def process_decode(p: processing.StableDiffusionProcessing, output):
if not hasattr(model, 'vae'):
if hasattr(model, 'pipe') and hasattr(model.pipe, 'vae'):
model = model.pipe
if hasattr(model, "vae") and output.images is not None and len(output.images) > 0:
if (hasattr(model, "vae") or hasattr(model, "vqgan")) and output.images is not None and len(output.images) > 0:
if p.hr_resize_mode > 0 and (p.hr_upscaler != 'None' or p.hr_resize_mode == 5):
width = max(getattr(p, 'width', 0), getattr(p, 'hr_upscale_to_x', 0))
height = max(getattr(p, 'height', 0), getattr(p, 'hr_upscale_to_y', 0))