From 501f19eec65aab44e425172b584e52409a096999 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Mon, 6 Mar 2023 11:34:02 -0500 Subject: [PATCH] Fix clip_skip no longer being loaded from yaml file. --- comfy/sd.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index 67a207cb..e19b2a35 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -266,7 +266,7 @@ class CLIP: self.cond_stage_model = clip(**(params)) self.tokenizer = tokenizer(embedding_directory=embedding_directory) self.patcher = ModelPatcher(self.cond_stage_model) - self.layer_idx = -1 + self.layer_idx = None def clone(self): n = CLIP(no_init=True) @@ -287,7 +287,8 @@ class CLIP: self.layer_idx = layer_idx def encode(self, text): - self.cond_stage_model.clip_layer(self.layer_idx) + if self.layer_idx is not None: + self.cond_stage_model.clip_layer(self.layer_idx) tokens = self.tokenizer.tokenize_with_weights(text) try: self.patcher.patch_model()