zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-25 18:32:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support latent previews for nested latents (#15196)

Browse Source
This commit is contained in:
Jukka Seppänen
2026-08-02 02:57:15 +03:00
committed by GitHub
parent e803f24ea0
commit 49a7422892
5 changed files with 153 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
comfy_extras/nodes_custom_sampler.py
View File

@@ -775,10 +775,11 @@ class SamplerCustom(io.ComfyNode):
out.pop("downscale_ratio_temporal", None)
out["samples"] = samples
if "x0" in x0_output:
x0_out = model.model.process_latent_out(x0_output["x0"].cpu())
if samples.is_nested:
x0 = x0_output["x0"]
if samples.is_nested and not x0.is_nested:
latent_shapes = [x.shape for x in samples.unbind()]
x0_out = comfy.nested_tensor.NestedTensor(comfy.utils.unpack_latents(x0_out, latent_shapes))
x0 = comfy.nested_tensor.NestedTensor(comfy.utils.unpack_latents(x0, latent_shapes))
x0_out = model.model.process_latent_out(x0.cpu())
out_denoised = latent.copy()
out_denoised["samples"] = x0_out
else:
@@ -1053,10 +1054,11 @@ class SamplerCustomAdvanced(io.ComfyNode):
out.pop("downscale_ratio_temporal", None)
out["samples"] = samples
if "x0" in x0_output:
x0_out = guider.model_patcher.model.process_latent_out(x0_output["x0"].cpu())
if samples.is_nested:
x0 = x0_output["x0"]
if samples.is_nested and not x0.is_nested:
latent_shapes = [x.shape for x in samples.unbind()]
x0_out = comfy.nested_tensor.NestedTensor(comfy.utils.unpack_latents(x0_out, latent_shapes))
x0 = comfy.nested_tensor.NestedTensor(comfy.utils.unpack_latents(x0, latent_shapes))
x0_out = guider.model_patcher.model.process_latent_out(x0.cpu())
out_denoised = latent.copy()
out_denoised["samples"] = x0_out
else: