1 files changed, 4 insertions, 3 deletions

diff --git a/infer.py b/infer.py
index 9b0ec1f..2bf9cb3 100644
--- a/infer.py
+++ b/infer.py
@@ -273,14 +273,15 @@ def generate(output_dir, pipeline, args):
 
             generator = torch.Generator(device="cuda").manual_seed(args.seed + i)
             images = pipeline(
-                prompt=args.prompt * (args.batch_size // len(args.prompt)),
+                prompt=args.prompt,
+                negative_prompt=args.negative_prompt,
                 height=args.height,
                 width=args.width,
-                negative_prompt=args.negative_prompt,
+                num_images_per_prompt=args.batch_size,
                 num_inference_steps=args.steps,
                 guidance_scale=args.guidance_scale,
                 generator=generator,
-                latents=init_image,
+                latents_or_image=init_image,
                 strength=args.image_noise,
             ).images