Upload 2 files

pipeline.py

@@ -1617,12 +1617,12 @@ class CustomPipeline(DiffusionPipeline, SD3LoraLoaderMixin):
                     pooled_prompt_embeds_input = torch.cat([negative_pooled_prompt_embeds, pooled_prompt_embeds], dim=0)
                     
                     # # Concatenate text IDs if they are used
-                    # if text_ids is not None and negative_text_ids is not None:
-                    #     text_ids_input = torch.cat([negative_text_ids, text_ids], dim=0)
+                    if text_ids is not None and negative_text_ids is not None:
+                        text_ids_input = torch.cat([negative_text_ids, text_ids], dim=0)
                     
                     # Concatenate latent image IDs if they are used
-                    # if latent_image_ids is not None:
-                    #     latent_image_ids_input = torch.cat([latent_image_ids, latent_image_ids], dim=0)
+                    if latent_image_ids is not None:
+                        latent_image_ids_input = torch.cat([latent_image_ids, latent_image_ids], dim=0)
                     
                     # Concatenate prompt masks if they are used
                     if prompt_mask is not None and negative_mask is not None: