diff --git a/modules_forge/forge_clip.py b/modules_forge/forge_clip.py index db814e31..9437d6b9 100644 --- a/modules_forge/forge_clip.py +++ b/modules_forge/forge_clip.py @@ -8,7 +8,7 @@ class CLIP_SD_15_L(FrozenCLIPEmbedderWithCustomWords): return super().encode_with_transformers(tokens) -class CLIP_SD_21_G(FrozenCLIPEmbedderWithCustomWords): +class CLIP_SD_21_H(FrozenCLIPEmbedderWithCustomWords): def __init__(self, wrapped, hijack): super().__init__(wrapped, hijack) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 4ce966c8..18bc5674 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -196,12 +196,12 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): sd_model.cond_stage_model = forge_clip.CLIP_SD_15_L(sd_model.cond_stage_model, sd_hijack.model_hijack) sd_model.cond_stage_model.patcher = forge_object.clip elif type(sd_model.cond_stage_model).__name__ == 'FrozenOpenCLIPEmbedder': # SD21 Clip - sd_model.cond_stage_model.tokenizer = forge_object.clip.tokenizer.clip_g.tokenizer - sd_model.cond_stage_model.transformer = forge_object.clip.cond_stage_model.clip_g.transformer + sd_model.cond_stage_model.tokenizer = forge_object.clip.tokenizer.clip_h.tokenizer + sd_model.cond_stage_model.transformer = forge_object.clip.cond_stage_model.clip_h.transformer model_embeddings = sd_model.cond_stage_model.transformer.text_model.embeddings model_embeddings.token_embedding = sd_hijack.EmbeddingsWithFixes( model_embeddings.token_embedding, sd_hijack.model_hijack) - sd_model.cond_stage_model = forge_clip.CLIP_SD_21_G(sd_model.cond_stage_model, sd_hijack.model_hijack) + sd_model.cond_stage_model = forge_clip.CLIP_SD_21_H(sd_model.cond_stage_model, sd_hijack.model_hijack) sd_model.cond_stage_model.patcher = forge_object.clip else: raise NotImplementedError('Bad Clip Class Name:' + type(sd_model.cond_stage_model).__name__)