mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-21 01:56:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make hook_scope functional for TransformerOptionsHook

Browse Source
This commit is contained in:
Jedrzej Kosinski
2025-01-06 02:23:04 -06:00
parent 6463c39ce0
commit f48f90e471
2 changed files with 28 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_patcher.py
View File

@@ -1010,11 +1010,11 @@ class ModelPatcher:
def apply_hooks(self, hooks: comfy.hooks.HookGroup, transformer_options: dict=None, force_apply=False):
# TODO: return transformer_options dict with any additions from hooks
if self.current_hooks == hooks and (not force_apply or (not self.is_clip and hooks is None)):
return {}
return comfy.hooks.create_transformer_options_from_hooks(self, hooks, transformer_options)
self.patch_hooks(hooks=hooks)
for callback in self.get_all_callbacks(CallbacksMP.ON_APPLY_HOOKS):
callback(self, hooks)
return {}
return comfy.hooks.create_transformer_options_from_hooks(self, hooks, transformer_options)
def patch_hooks(self, hooks: comfy.hooks.HookGroup):
with self.use_ejected():