Update pipeline_stable_diffusion_3_ipa.py

pipeline_stable_diffusion_3_ipa.py

@@ -1191,9 +1191,10 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
         if linear_layer.weight.device.type == 'cpu':
             linear_layer.to('cuda')
         '''
-
-        batch_size = concatenated_embeds.shape[0]
-        reshaped_embeds = concatenated_embeds.view(batch_size, input_dim)
+        bs_embed, seq_len, _ = concatenated_embeds.shape
+       # batch_size = concatenated_embeds.shape[0]
+        image_prompt_embeds = image_prompt_embeds.repeat(1, 1, 1)
+        reshaped_embeds = concatenated_embeds.view(bs_embed, seq_len, -1)
         # Pass the concatenated embeddings through the linear layer
         clip_image_embeds = linear_layer(reshaped_embeds)