Commit 501f19ee authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix clip_skip no longer being loaded from yaml file.

parent afff30fc
......@@ -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,6 +287,7 @@ class CLIP:
self.layer_idx = layer_idx
def encode(self, text):
if self.layer_idx is not None:
self.cond_stage_model.clip_layer(self.layer_idx)
tokens = self.tokenizer.tokenize_with_weights(text)
try:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment