From b7933960bbc5967864a5da93377def624aca8f97 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Sat, 24 Jun 2023 13:56:46 -0400 Subject: [PATCH] Fix CLIPLoader node. --- comfy/sd.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/comfy/sd.py b/comfy/sd.py index 74c144b..6feb0de 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -961,12 +961,19 @@ def load_style_model(ckpt_path): def load_clip(ckpt_path, embedding_directory=None): clip_data = utils.load_torch_file(ckpt_path, safe_load=True) - config = {} + class EmptyClass: + pass + + clip_target = EmptyClass() + clip_target.params = {} if "text_model.encoder.layers.22.mlp.fc1.weight" in clip_data: - config['target'] = 'comfy.ldm.modules.encoders.modules.FrozenOpenCLIPEmbedder' + clip_target.clip = sd2_clip.SD2ClipModel + clip_target.tokenizer = sd2_clip.SD2Tokenizer else: - config['target'] = 'comfy.ldm.modules.encoders.modules.FrozenCLIPEmbedder' - clip = CLIP(config=config, embedding_directory=embedding_directory) + clip_target.clip = sd1_clip.SD1ClipModel + clip_target.tokenizer = sd1_clip.SD1Tokenizer + + clip = CLIP(clip_target, embedding_directory=embedding_directory) clip.load_from_state_dict(clip_data) return clip