Skip to content

GitLab

Commit 1f4fc9ea authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix issue with get_model_object on patched model.

parent 1a0486bb
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
comfy/model_patcher.py
View file @ 1f4fc9ea
...@@ -153,6 +153,9 @@ class ModelPatcher: ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment