diff --git a/nodes/WASHybridLatentUpscale.py b/nodes/WASHybridLatentUpscale.py index 081750d..a54762b 100644 --- a/nodes/WASHybridLatentUpscale.py +++ b/nodes/WASHybridLatentUpscale.py @@ -322,7 +322,7 @@ class WASLatentUpscaleHybrid: donor_latent=donor_samples, ) - out = dict(samples) + out = dict(latent) out["samples"] = up_latent if cfg.output_mask_resolution == "latent":