From 6ea2651e7df66d7585f6ffee804b20e92fb38b8a Mon Sep 17 00:00:00 2001 From: City <125218114+city96@users.noreply.github.com> Date: Mon, 12 Jan 2026 13:55:28 +0100 Subject: [PATCH] Only include metadata on new comfy versions For #407 since old comfy versions don't support passing metadata (added in #399 ) --- nodes.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index 50d3a60..4683514 100644 --- a/nodes.py +++ b/nodes.py @@ -1,6 +1,7 @@ # (c) City96 || Apache-2.0 (apache.org/licenses/LICENSE-2.0) import torch import logging +import inspect import collections import nodes @@ -166,8 +167,14 @@ class UnetLoaderGGUF: # init model unet_path = folder_paths.get_full_path("unet", unet_name) sd, extra = gguf_sd_loader(unet_path) + + kwargs = {} + valid_params = inspect.signature(comfy.sd.load_diffusion_model_state_dict).parameters + if "metadata" in valid_params: + kwargs["metadata"] = extra.get("metadata", {}) + model = comfy.sd.load_diffusion_model_state_dict( - sd, model_options={"custom_operations": ops}, metadata=extra.get("metadata", {}) + sd, model_options={"custom_operations": ops}, **kwargs, ) if model is None: logging.error("ERROR UNSUPPORTED UNET {}".format(unet_path))