Traceback (most recent call last):
File “/content/train_text_to_image_lora_sdxl.py”, line 1327, in
main(args)
File “/content/train_text_to_image_lora_sdxl.py”, line 877, in main
column_names = dataset[“train”].column_names
File “/usr/local/lib/python3.10/dist-packages/datasets/dataset_dict.py”, line 75, in getitem
return super().getitem(k)
KeyError: ‘train’
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 48, in main
args.func(args)
File “/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py”, line 1097, in launch_command
simple_launcher(args)
File “/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py”, line 703, in simple_launcher
raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)
subprocess.CalledProcessError: Command ‘[’/usr/bin/python3’, ‘train_text_to_image_lora_sdxl.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=data’, ‘–output_dir=output_lora’, ‘–caption_column=prompt’, ‘–mixed_precision=fp16’, ‘–validation_prompt=a man with a rose in his mouth’, ‘–resolution=1024’, ‘–train_batch_size=1’, ‘–gradient_accumulation_steps=1’, ‘–gradient_checkpointing’, ‘–learning_rate=1e-4’, ‘–snr_gamma=5.0’, ‘–lr_scheduler=constant’, ‘–lr_warmup_steps=0’, ‘–push_to_hub’, ‘–mixed_precision=fp16’, ‘–use_8bit_adam’, ‘–max_train_steps=500’, ‘–checkpointing_steps=717’, ‘–seed=0’]’ returned non-zero exit status 1.