From 6c5b5b260eddfa750bae5757043ac08d8fc5503c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rastislav=20=C5=A0varba?= Date: Wed, 9 Aug 2023 08:18:24 +0200 Subject: [PATCH] Fix push_to_hub in train_text_to_image_lora_sdxl.py example (#4535) fix: push_to_hub in train text2image lora sdxl --- examples/text_to_image/train_text_to_image_lora_sdxl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/text_to_image/train_text_to_image_lora_sdxl.py b/examples/text_to_image/train_text_to_image_lora_sdxl.py index b5e7ee3e7d..b93c4327bb 100644 --- a/examples/text_to_image/train_text_to_image_lora_sdxl.py +++ b/examples/text_to_image/train_text_to_image_lora_sdxl.py @@ -1262,8 +1262,8 @@ def main(args): repo_id, images=images, base_model=args.pretrained_model_name_or_path, + dataset_name=args.dataset_name, train_text_encoder=args.train_text_encoder, - prompt=args.instance_prompt, repo_folder=args.output_dir, vae_path=args.pretrained_vae_model_name_or_path, )