Update pipeline.py

pipeline.py

@@ -854,18 +854,16 @@ class LibreFluxControlNetPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSi
                     current_prompt_embeds = torch.cat([negative_prompt_embeds, prompt_embeds])
                     current_pooled_embeds = torch.cat([negative_pooled_prompt_embeds, pooled_prompt_embeds])
                     current_attention_mask = torch.cat([negative_attention_mask, attention_mask])
-                    print('Neg Text Ids Shape:',negative_text_ids.shape,'Text Ids Shape:', text_ids.shape, 'Latent Image Ids Shape:',latent_image_ids.shape)
-
-                    current_text_ids = torch.cat([negative_text_ids, text_ids])
-                    current_img_ids = torch.cat([latent_image_ids] * 2)
+                    current_text_ids = text_ids[0]
+                    current_img_ids = latent_image_ids[0]
                     current_control_image = torch.cat([control_image] * 2) if isinstance(control_image, torch.Tensor) else [torch.cat([c_img] * 2) for c_img in control_image]
                 else:
                     latent_model_input = latents
                     current_prompt_embeds = prompt_embeds
                     current_pooled_embeds = pooled_prompt_embeds
                     current_attention_mask = attention_mask
-                    current_text_ids = text_ids
-                    current_img_ids = latent_image_ids
+                    current_text_ids = text_ids[0]
+                    current_img_ids = latent_image_ids[0]
                     current_control_image = control_image
 
                 # FIX: Integrate with device handling