From f48c0c15c98a6e8e0eb0f5d44eaec353f4fb4d53 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Fri, 29 Nov 2024 18:42:34 -0600 Subject: [PATCH] Make sure encode_from_tokens_scheduled will respect use_clip_schedule on clip --- comfy/sd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index 8db83ce82..ebae7f996 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -137,8 +137,8 @@ class CLIP: def encode_from_tokens_scheduled(self, tokens, unprojected=False, add_dict: dict[str]={}, show_pbar=True): all_cond_pooled: list[tuple[torch.Tensor, dict[str]]] = [] all_hooks = self.patcher.forced_hooks - if all_hooks is None: - # if no hooks, do unscheduled encode_from_tokens and perform add_dict + if all_hooks is None or not self.use_clip_schedule: + # if no hooks or shouldn't use clip schedule, do unscheduled encode_from_tokens and perform add_dict return_pooled = "unprojected" if unprojected else True pooled_dict = self.encode_from_tokens(tokens, return_pooled=return_pooled, return_dict=True) cond = pooled_dict.pop("cond")