Dreambooth: Support loading Textual Inversion embeddings

1 files changed, 1 insertions, 1 deletions

diff --git a/infer.py b/infer.py
index e3fa9e5..5bd926a 100644
--- a/infer.py
+++ b/infer.py
@@ -291,7 +291,7 @@ def generate(output_dir, pipeline, args):
                 num_inference_steps=args.steps,
                 guidance_scale=args.guidance_scale,
                 generator=generator,
-                latents_or_image=init_image,
+                image=init_image,
                 strength=args.image_noise,
             ).images