Skip to content

Commit 6ea2651

Browse files
committed
Only include metadata on new comfy versions
For #407 since old comfy versions don't support passing metadata (added in #399 )
1 parent 9ecc3c4 commit 6ea2651

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

nodes.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# (c) City96 || Apache-2.0 (apache.org/licenses/LICENSE-2.0)
22
import torch
33
import logging
4+
import inspect
45
import collections
56

67
import nodes
@@ -166,8 +167,14 @@ def load_unet(self, unet_name, dequant_dtype=None, patch_dtype=None, patch_on_de
166167
# init model
167168
unet_path = folder_paths.get_full_path("unet", unet_name)
168169
sd, extra = gguf_sd_loader(unet_path)
170+
171+
kwargs = {}
172+
valid_params = inspect.signature(comfy.sd.load_diffusion_model_state_dict).parameters
173+
if "metadata" in valid_params:
174+
kwargs["metadata"] = extra.get("metadata", {})
175+
169176
model = comfy.sd.load_diffusion_model_state_dict(
170-
sd, model_options={"custom_operations": ops}, metadata=extra.get("metadata", {})
177+
sd, model_options={"custom_operations": ops}, **kwargs,
171178
)
172179
if model is None:
173180
logging.error("ERROR UNSUPPORTED UNET {}".format(unet_path))

0 commit comments

Comments
 (0)