From f28d892c16e3cf1fd35a5cd35cb5d7cf49e13969 Mon Sep 17 00:00:00 2001 From: "kosinkadink1@gmail.com" Date: Sat, 21 Sep 2024 16:34:40 +0900 Subject: [PATCH] Fix skip_until_exit logic bug breaking injection after first run of model --- comfy/model_patcher.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index f5dc81b33..57d7353e1 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -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