notebooks icon indicating copy to clipboard operation
notebooks copied to clipboard

SDXL_Dreambooth_LoRA_advanced_example.ipynb Training Issue

Open mhy-666 opened this issue 1 year ago • 0 comments
trafficstars

Hellom, when I used SDXL_Dreambooth_LoRA_advanced_example.ipynb in colab to train my Lora model using my own dataset, it shows this error:

train_dreambooth_lora_sdxl_advanced.py: error: unrecognized arguments: cartoon girl cartoon girl is reading a book Traceback (most recent call last): File "/usr/local/bin/accelerate", line 8, in sys.exit(main()) File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/accelerate_cli.py", line 47, in main args.func(args) File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py", line 1017, in launch_command simple_launcher(args) File "/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py", line 637, in simple_launcher raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) subprocess.CalledProcessError: Command '['/usr/bin/python3', 'train_dreambooth_lora_sdxl_advanced.py', '--pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0', '--pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix', '--dataset_name=./yucca', '--instance_prompt=yucca', 'cartoon', 'girl', '--validation_prompt=yucca', 'cartoon', 'girl', 'is', 'reading', 'a', 'book', '--output_dir=yucca-sdxl-lora', '--caption_column=prompt', '--mixed_precision=bf16', '--resolution=1024', '--train_batch_size=3', '--repeats=1', '--report_to=wandb', '--gradient_accumulation_steps=1', '--gradient_checkpointing', '--learning_rate=1.0', '--text_encoder_lr=1.0', '--adam_beta2=0.99', '--optimizer=prodigy', '--train_text_encoder_ti', '--train_text_encoder_ti_frac=0.5', '--snr_gamma=5.0', '--lr_scheduler=constant', '--lr_warmup_steps=0', '--rank=8', '--max_train_steps=1000', '--checkpointing_steps=2000', '--seed=0', '--push_to_hub']' returned non-zero exit status 2.

It seems that something wrong with parameters--instance_prompt=$instance_prompt \and --validation_prompt=$validation_prompt \ , how can I deal with this situation?

mhy-666 avatar Jan 10 '24 04:01 mhy-666