From bb9f33eb6293ac18a09b65ecda5b137d6bf6d96e Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 04:57:02 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 9d555bb6..71027236 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -162,7 +162,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): typename = type(embedder).__name__ if typename == 'FrozenOpenCLIPEmbedder': embedder.tokenizer = forge_object.clip.tokenizer.clip_g.tokenizer - embedder.model = forge_object.clip.cond_stage_model.clip_g.transformer + embedder.transformer = forge_object.clip.cond_stage_model.clip_g.transformer text_cond_models.append(embedder) elif typename == 'FrozenCLIPEmbedder': embedder.tokenizer = forge_object.clip.tokenizer.clip_l.tokenizer @@ -170,7 +170,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): text_cond_models.append(embedder) elif typename == 'FrozenOpenCLIPEmbedder2': embedder.tokenizer = forge_object.clip.tokenizer.clip_g.tokenizer - embedder.model = forge_object.clip.cond_stage_model.clip_g.transformer + embedder.transformer = forge_object.clip.cond_stage_model.clip_g.transformer text_cond_models.append(embedder) if len(text_cond_models) == 1: @@ -182,7 +182,7 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): sd_model.cond_stage_model.transformer = forge_object.clip.cond_stage_model.clip_l.transformer elif type(sd_model.cond_stage_model).__name__ == 'FrozenOpenCLIPEmbedder': sd_model.cond_stage_model.tokenizer = forge_object.clip.tokenizer.clip_g.tokenizer - sd_model.cond_stage_model.model = forge_object.clip.cond_stage_model.clip_g.transformer + sd_model.cond_stage_model.transformer = forge_object.clip.cond_stage_model.clip_g.transformer else: raise NotImplementedError('Bad Clip Class Name:' + type(sd_model.cond_stage_model).__name__)