298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix clip_skip no longer being loaded from yaml file.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-06 11:34:02 -05:00
parent afff30fc0a
commit 501f19eec6
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/sd.py
View File

@ -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()