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"]