Update comfy_extras/nodes_seedvr.py

This commit is contained in:
Alexis Rolland 2026-06-08 17:43:07 +08:00 committed by GitHub
parent 7dcea515ef
commit 09e6cd4689
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -268,8 +268,8 @@ class SeedVR2PostProcessing(io.ComfyNode):
category="image/upscaling", category="image/upscaling",
description="Align the generated image with the original resized image and apply color correction.", description="Align the generated image with the original resized image and apply color correction.",
inputs=[ inputs=[
io.Image.Input("images", tooltip="The decoded upscaled image to color-correct."), io.Image.Input("images", tooltip="The generated image to process."),
io.Image.Input("original_resized_images", tooltip="The original resized (pre-pad) image used as the geometry and color reference."), io.Image.Input("original_resized_images", tooltip="The original resized image before pre-processing, used as reference."),
io.Combo.Input("color_correction_method", options=["lab", "wavelet", "adain", "none"], default="lab", tooltip="How to match the output's color to the original. lab: transfer color in CIELAB space, preserving detail (most faithful). wavelet: transfer low-frequency color, keeping upscaled high-frequency detail. adain: match per-channel mean/std (fastest, global tint). none: skip color transfer (geometry alignment only)."), io.Combo.Input("color_correction_method", options=["lab", "wavelet", "adain", "none"], default="lab", tooltip="How to match the output's color to the original. lab: transfer color in CIELAB space, preserving detail (most faithful). wavelet: transfer low-frequency color, keeping upscaled high-frequency detail. adain: match per-channel mean/std (fastest, global tint). none: skip color transfer (geometry alignment only)."),
], ],
outputs=[io.Image.Output(display_name="images")], outputs=[io.Image.Output(display_name="images")],