Unverified Commit 6c5b5b26 authored by Rastislav Švarba's avatar Rastislav Švarba Committed by GitHub
Browse files

Fix push_to_hub in train_text_to_image_lora_sdxl.py example (#4535)

fix: push_to_hub in train text2image lora sdxl
parent e7e37494
...@@ -1262,8 +1262,8 @@ def main(args): ...@@ -1262,8 +1262,8 @@ def main(args):
repo_id, repo_id,
images=images, images=images,
base_model=args.pretrained_model_name_or_path, base_model=args.pretrained_model_name_or_path,
dataset_name=args.dataset_name,
train_text_encoder=args.train_text_encoder, train_text_encoder=args.train_text_encoder,
prompt=args.instance_prompt,
repo_folder=args.output_dir, repo_folder=args.output_dir,
vae_path=args.pretrained_vae_model_name_or_path, vae_path=args.pretrained_vae_model_name_or_path,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment