Update pipeline.py

pipeline.py

@@ -182,11 +182,11 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
             )
         prompt_embeds = self.text_encoder(text_input_ids.to(device), output_hidden_states=False)
 
-        _, seq_len, _ = prompt_embeds.shape
-
         # Use pooled output of CLIPTextModel
         prompt_embeds = prompt_embeds.pooler_output
         prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
+        
+        _, seq_len, _ = prompt_embeds.shape
 
         # duplicate text embeddings for each generation per prompt, using mps friendly method
         prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt)