Update src/diffusers/pipelines/longcat_image/pipeline_longcat_image.py

junqiangwu · yiyixuxu · web-flow · commit 5e2f36c5a221 · 2025-12-13T03:46:59.000+08:00
Co-authored-by: YiYi Xu &lt;yixu310@gmail.com&gt;
diff --git a/src/diffusers/pipelines/longcat_image/pipeline_longcat_image.py b/src/diffusers/pipelines/longcat_image/pipeline_longcat_image.py
@@ -334,6 +334,8 @@ def _encode_prompt( self, prompt ):
         )
         # [max_sequence_length, batch, hidden_size] -> [batch, max_sequence_length, hidden_size]
         # clone to have a contiguous tensor
+        self.prompt_template_encode_start_idx = len(prefix_tokens)
+        self.prompt_template_encode_end_idx = len(suffix_tokens)
         prompt_embeds = text_output.hidden_states[-1].detach()
         prompt_embeds = prompt_embeds[:,self.prompt_template_encode_start_idx: -self.prompt_template_encode_end_idx ,:]
 

Original file line number	Diff line number	Diff line change
`@@ -334,6 +334,8 @@ def _encode_prompt( self, prompt ):`
`334`	`334`	`)`
`335`	`335`	`# [max_sequence_length, batch, hidden_size] -> [batch, max_sequence_length, hidden_size]`
`336`	`336`	`# clone to have a contiguous tensor`
	`337`	`+ self.prompt_template_encode_start_idx = len(prefix_tokens)`
	`338`	`+ self.prompt_template_encode_end_idx = len(suffix_tokens)`
`337`	`339`	`prompt_embeds = text_output.hidden_states[-1].detach()`
`338`	`340`	`prompt_embeds = prompt_embeds[:,self.prompt_template_encode_start_idx: -self.prompt_template_encode_end_idx ,:]`
`339`	`341`