Got keyerror train error while fine tuning stability ai model using LORA
Traceback (most recent call last):File “/content/train_text_to_image_lora_sdxl.py”, line 1327, inmain(args)File “/content/train_text_to_image_lora_sdxl.py”, line 877, in maincolumn_names = dataset[“train”].column_namesFile “/usr/local/lib/python3.10/dist-packages/datasets/dataset_dict.py”, line 75, ingetitemreturn super().getitem(k)KeyError: ‘train’Traceback (most recent call last):File “/usr/local/bin/accelerate”, line 8, insys.exit(main())File “/usr/local/lib/python3.10/dist-packages/accelerate/commands/accelerate_cli.py”, line 48, in mainargs.func(args)File “/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py”, line 1097, in launch_commandsimple_launcher(args)File “/usr/local/lib/python3.10/dist-packages/accelerate/commands/launch.py”, line 703, in simple_launcherraise 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.