From 005c57839c5bee88f1a0a41970ca965ab470a4c8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=A1=E5=9D=82=E6=A1=9C?= Date: Tue, 1 Sep 2026 11:21:44 +0800 Subject: [PATCH] fix: use native VAEDecodeTiled for tiled decode (Qwen Image VAE support) (#1028) --- py/nodes/samplers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/py/nodes/samplers.py b/py/nodes/samplers.py index c8ac6dc..d00dd44 100644 --- a/py/nodes/samplers.py +++ b/py/nodes/samplers.py @@ -8,7 +8,7 @@ import comfy_extras.nodes_custom_sampler as custom_samplers from tqdm import trange from server import PromptServer -from nodes import RepeatLatentBatch, NODE_CLASS_MAPPINGS as ALL_NODE_CLASS_MAPPINGS, VAEEncodeForInpaint, InpaintModelConditioning +from nodes import RepeatLatentBatch, NODE_CLASS_MAPPINGS as ALL_NODE_CLASS_MAPPINGS, VAEEncodeForInpaint, InpaintModelConditioning, VAEDecodeTiled from ..modules.layer_diffuse import LayerDiffuse from ..config import * @@ -362,7 +362,7 @@ class samplerFull: spent_time = 'Diffusion:' + str((end_time - start_time) / 1000) + '″' else: if tile_size is not None: - samp_images = samp_vae.decode_tiled(latent, tile_x=tile_size // 8, tile_y=tile_size // 8, ) + samp_images, = VAEDecodeTiled().decode(samp_vae, {"samples": latent}, tile_size) else: samp_images = samp_vae.decode(latent).cpu() if len(samp_images.shape) == 5: # Combine batches @@ -980,7 +980,7 @@ class samplerSDTurbo: # 解码图片 if tile_size is not None: - samp_images = samp_vae.decode_tiled(latent, tile_x=tile_size // 8, tile_y=tile_size // 8, ) + samp_images, = VAEDecodeTiled().decode(samp_vae, {"samples": latent}, tile_size) else: samp_images = samp_vae.decode(latent).cpu()