From f4a91a640ce140f7d5b2f7e4a9cd015aeeaddb22 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Mon, 26 Sep 2022 14:37:32 +0200 Subject: [PATCH] fix key name --- examples/dreambooth/train_dreambooth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 7b95f21963..2959b57231 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -434,7 +434,7 @@ def main(): batch = { "instance_images": instance_images, - "input_ids": instance_prompt_ids, + "instance_prompt_ids": instance_prompt_ids, } if args.with_prior_preservation: