diff --git a/modules/prompt_parser_diffusers.py b/modules/prompt_parser_diffusers.py index 1b356d7e9..2025a4ca6 100644 --- a/modules/prompt_parser_diffusers.py +++ b/modules/prompt_parser_diffusers.py @@ -327,7 +327,9 @@ def compel_hijack(self, token_ids: torch.Tensor, attention_mask: torch.Tensor | else: hidden_state = text_encoder_output.hidden_states[-clip_skip] if normalized: - hidden_state = self.text_encoder.text_model.final_layer_norm(hidden_state) + # transformers >=5.6 flattened CLIPTextModel; CLIPTextModelWithProjection still nests it under .text_model + text_model = getattr(self.text_encoder, 'text_model', self.text_encoder) + hidden_state = text_model.final_layer_norm(hidden_state) return hidden_state diff --git a/modules/prompt_parser_xhinker.py b/modules/prompt_parser_xhinker.py index 344f0778c..1eeefe88d 100644 --- a/modules/prompt_parser_xhinker.py +++ b/modules/prompt_parser_xhinker.py @@ -216,9 +216,11 @@ def get_weighted_text_embeddings_sd15( , generator = torch.Generator(text2img_pipe.device).manual_seed(2) ).images[0] """ - original_clip_layers = pipe.text_encoder.text_model.encoder.layers + # transformers >=5.6 flattened CLIPTextModel; CLIPTextModelWithProjection still nests it under .text_model + clip_text_model = getattr(pipe.text_encoder, 'text_model', pipe.text_encoder) + original_clip_layers = clip_text_model.encoder.layers if clip_skip > 0: - pipe.text_encoder.text_model.encoder.layers = original_clip_layers[:-clip_skip] + clip_text_model.encoder.layers = original_clip_layers[:-clip_skip] eos = pipe.tokenizer.eos_token_id prompt_tokens, prompt_weights = get_prompts_tokens_with_weights( @@ -310,7 +312,7 @@ def get_weighted_text_embeddings_sd15( # recover clip layers if clip_skip > 0: - pipe.text_encoder.text_model.encoder.layers = original_clip_layers + clip_text_model.encoder.layers = original_clip_layers return prompt_embeds, neg_prompt_embeds