fix bug

pipeline.py

@@ -646,7 +646,7 @@ class StableDiffusionLongPromptWeightingPipeline(DiffusionPipeline):
             init_latent_dist = self.vae.encode(init_image).latent_dist
             init_latents = init_latent_dist.sample(generator=generator)
             init_latents = 0.18215 * init_latents
-            init_latents = torch.cat([init_latents] * num_images_per_prompt, dim=0)
+            init_latents = torch.cat([init_latents] * batch_size * num_images_per_prompt, dim=0)
             init_latents_orig = init_latents
 
             # preprocess mask