Update pipeline_stable_diffusion_3_ipa.py

pipeline_stable_diffusion_3_ipa.py

@@ -1189,8 +1189,11 @@ 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)
         # Pass the concatenated embeddings through the linear layer
-        clip_image_embeds = linear_layer(concatenated_embeds)
+        clip_image_embeds = linear_layer(reshaped_embeds)
 
         # Add a ReLU activation for non-linearity (optional)
         #combined_embeds = torch.relu(combined_embeds)