From 48a814315f500a6f3e87ac4c8446305f8b2ef8f6 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Sun, 27 Jul 2025 15:53:40 +0900 Subject: [PATCH] modified: detailer - allow 5D latent model (WAN) - assuming it's a single-frame latent, not a video latent https://github.com/ltdrdata/ComfyUI-Impact-Pack/issues/1066 --- modules/impact/config.py | 2 +- modules/impact/core.py | 5 +++++ pyproject.toml | 2 +- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/modules/impact/config.py b/modules/impact/config.py index c5f57ba..0218ca3 100644 --- a/modules/impact/config.py +++ b/modules/impact/config.py @@ -3,7 +3,7 @@ import os import logging -version_code = [8, 21, 1] +version_code = [8, 21, 2] version = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '') my_path = os.path.dirname(__file__) diff --git a/modules/impact/core.py b/modules/impact/core.py index 774d1f9..f14c322 100644 --- a/modules/impact/core.py +++ b/modules/impact/core.py @@ -407,6 +407,11 @@ def enhance_detail(image, model, clip, vae, guide_size, guide_size_for_bbox, max refined_image = detailer_hook.post_decode(refined_image) # downscale + + # workaround: support WAN as an i2i model + if len(refined_image.shape) == 5: + refined_image = refined_image.squeeze(0) + refined_image = utils.tensor_resize(refined_image, w, h) # prevent mixing of device diff --git a/pyproject.toml b/pyproject.toml index 7279c4c..57cf164 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-impact-pack" description = "This node pack offers various detector nodes and detailer nodes that allow you to configure a workflow that automatically enhances facial details. And provide iterative upscaler." -version = "8.21.1" +version = "8.21.2" license = { file = "LICENSE.txt" } dependencies = ["segment-anything", "scikit-image", "piexif", "transformers", "opencv-python-headless", "GitPython", "scipy>=1.11.4"]