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

Fix skip_until_exit logic bug breaking injection after first run of model

Browse Source
This commit is contained in:
kosinkadink1@gmail.com 2024-09-21 16:34:40 +09:00
parent 5f450d3351
commit f28d892c16
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_patcher.py
View File

@ -180,7 +180,10 @@ class AutoPatcherEjector:
def __exit__(self, *args):
if self.was_injected:
if self.skip_until_exit or not self.model.skip_injection:
if self.skip_until_exit:
self.model.skip_injection = self.prev_skip_injection
self.model.inject_model()
elif not self.model.skip_injection:
self.model.inject_model()
self.model.skip_injection = self.prev_skip_injection