diff --git a/comfy_extras/nodes_model_patch.py b/comfy_extras/nodes_model_patch.py index a9235af60..b892d7cde 100644 --- a/comfy_extras/nodes_model_patch.py +++ b/comfy_extras/nodes_model_patch.py @@ -259,7 +259,7 @@ class ModelPatchLoader: model = comfy.ldm.lumina.controlnet.ZImage_Control(device=comfy.model_management.unet_offload_device(), dtype=dtype, operations=comfy.ops.manual_cast, **config) model_patcher = comfy.model_patcher.CoreModelPatcher(model, load_device=comfy.model_management.get_torch_device(), offload_device=comfy.model_management.unet_offload_device()) - model.load_state_dict(sd, assign=self.model_patcher.is_dynamic()) + model.load_state_dict(sd, assign=model_patcher.is_dynamic()) return (model_patcher,)