Browse Source

Fix issue with get_model_object on patched model.

pull/3216/head
comfyanonymous 8 months ago
parent
commit
1f4fc9ea0c
  1. 3
      comfy/model_patcher.py

3
comfy/model_patcher.py

@ -153,6 +153,9 @@ class ModelPatcher:
def get_model_object(self, name): def get_model_object(self, name):
if name in self.object_patches: if name in self.object_patches:
return self.object_patches[name] return self.object_patches[name]
else:
if name in self.object_patches_backup:
return self.object_patches_backup[name]
else: else:
return comfy.utils.get_attr(self.model, name) return comfy.utils.get_attr(self.model, name)

Loading…
Cancel
Save