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: