Update pipeline_stable_diffusion_3_ipa.py

pipeline_stable_diffusion_3_ipa.py

@@ -1217,7 +1217,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
         print('shape 2a: ', average_image_embedf.shape)
         # 3. Create a tensor of zeros with the same shape as the averaged embedding
         zeros_tensor = torch.zeros_like(average_image_embed)
-        zeros_tensora = torch.zeros_like(average_image_embed)
+        zeros_tensora = torch.zeros_like(average_image_embed).unsqueeze(0) 
         print('shape 3: ', zeros_tensor.shape)
         print('shape 3a: ', zeros_tensora.shape)
         # 4. Concatenate the zeros and the average embedding