xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-01 18:47:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with get_model_object on patched model.

Browse Source
This commit is contained in:
comfyanonymous 2024-04-04 23:01:02 -04:00
parent 22462103f3
commit 50463b3e5a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_patcher.py
View File

@ -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)