diff --git a/comfy_api/latest/_io.py b/comfy_api/latest/_io.py index 863254ce7..79c0722a9 100644 --- a/comfy_api/latest/_io.py +++ b/comfy_api/latest/_io.py @@ -629,6 +629,10 @@ class UpscaleModel(ComfyTypeIO): if TYPE_CHECKING: Type = ImageModelDescriptor +@comfytype(io_type="LATENT_UPSCALE_MODEL") +class LatentUpscaleModel(ComfyTypeIO): + Type = Any + @comfytype(io_type="AUDIO") class Audio(ComfyTypeIO): class AudioDict(TypedDict): diff --git a/comfy_extras/nodes_hunyuan.py b/comfy_extras/nodes_hunyuan.py index acd225f21..536c0c23f 100644 --- a/comfy_extras/nodes_hunyuan.py +++ b/comfy_extras/nodes_hunyuan.py @@ -176,10 +176,10 @@ class LatentUpscaleModelLoader(io.ComfyNode): display_name="Load Latent Upscale Model", category="loaders", inputs=[ - io.Combo.Input("model_name", options=folder_paths.get_filename_list("upscale_models")), + io.Combo.Input("model_name", options=folder_paths.get_filename_list("latent_upscale_models")), ], outputs=[ - io.UpscaleModel.Output(), + io.LatentUpscaleModel.Output(), ], ) @@ -220,7 +220,7 @@ class HunyuanVideo15LatentUpscaleWithModel(io.ComfyNode): display_name="Hunyuan Video 15 Latent Upscale With Model", category="latent", inputs=[ - io.UpscaleModel.Input("model"), + io.LatentUpscaleModel.Input("model"), io.Latent.Input("samples"), io.Combo.Input("upscale_method", options=["nearest-exact", "bilinear", "area", "bicubic", "bislerp"], default="bilinear"), io.Int.Input("width", default=1280, min=0, max=16384, step=8), diff --git a/folder_paths.py b/folder_paths.py index f110d832b..ffdc4d020 100644 --- a/folder_paths.py +++ b/folder_paths.py @@ -38,6 +38,8 @@ folder_names_and_paths["gligen"] = ([os.path.join(models_dir, "gligen")], suppor folder_names_and_paths["upscale_models"] = ([os.path.join(models_dir, "upscale_models")], supported_pt_extensions) +folder_names_and_paths["latent_upscale_models"] = ([os.path.join(models_dir, "latent_upscale_models")], supported_pt_extensions) + folder_names_and_paths["custom_nodes"] = ([os.path.join(base_path, "custom_nodes")], set()) folder_names_and_paths["hypernetworks"] = ([os.path.join(models_dir, "hypernetworks")], supported_pt_extensions) diff --git a/models/latent_upscale_models/put_latent_upscale_models_here b/models/latent_upscale_models/put_latent_upscale_models_here new file mode 100644 index 000000000..e69de29bb