xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-10-26 06:06:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix incorrect ref to create_hook_patches_clone after moving function

Browse Source
This commit is contained in:
kosinkadink1@gmail.com 2024-09-19 11:57:19 +09:00
parent 787ef34842
commit e80dc96627
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_patcher.py
View File

@ -202,8 +202,8 @@ class ModelPatcher:
for k, c in self.callbacks.items(): for k, c in self.callbacks.items():
n.callbacks[k] = c.copy() n.callbacks[k] = c.copy()
# hooks # hooks
n.hook_patches = self.create_hook_patches_clone(self.hook_patches) n.hook_patches = create_hook_patches_clone(self.hook_patches)
n.hook_patches_backup = self.create_hook_patches_clone(self.hook_patches_backup) n.hook_patches_backup = create_hook_patches_clone(self.hook_patches_backup)
# TODO: do we really need to clone cached_hook_patches/current_hooks? # TODO: do we really need to clone cached_hook_patches/current_hooks?
for group in self.cached_hook_patches: for group in self.cached_hook_patches:
n.cached_hook_patches[group] = {} n.cached_hook_patches[group] = {}
@ -678,7 +678,7 @@ class ModelPatcher:
if hook.hook_ref not in self.hook_patches: if hook.hook_ref not in self.hook_patches:
weight_hooks_to_register.append(hook) weight_hooks_to_register.append(hook)
if len(weight_hooks_to_register) > 0: if len(weight_hooks_to_register) > 0:
self.hook_patches_backup = self.create_hook_patches_clone(self.hook_patches) self.hook_patches_backup = create_hook_patches_clone(self.hook_patches)
for hook in weight_hooks_to_register: for hook in weight_hooks_to_register:
hook.add_hook_patches(self, target) hook.add_hook_patches(self, target)
for callback in self.callbacks[CallbacksMP.ON_REGISTER_ALL_HOOK_PATCHES]: for callback in self.callbacks[CallbacksMP.ON_REGISTER_ALL_HOOK_PATCHES]: