instance_images = [Image.open(path) for path in list(Path(instance_data_root).iterdir())]
# 过滤掉目录,只保留文件
instance_images = [Image.open(path) for path in list(Path(instance_data_root).iterdir()) if path.is_file()]
# 或者只处理指定格式的图像文件
instance_images = [Image.open(path) for path in list(Path(instance_data_root).iterdir()) if path.is_file() and path.suffix in ['.jpg', '.jpeg', '.png']]
复制代码
以下报错可以忽略
subprocess.CalledProcessError: Command '['./dreambooth_env/bin/python', 'train_dreambooth_lora_sdxl.py', '--pretrained_moth=stabilityai/stable-diffusion-xl-base-1.0', '--instance_data_dir=dog', '--pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16ut_dir=lora-trained-xl', '--mixed_precision=fp16', '--instance_prompt=a photo of sks dog', '--resolution=1024', '--train_batch_size=1',ccumulation_steps=4', '--learning_rate=1e-4', '--report_to=wandb', '--lr_scheduler=constant', '--lr_warmup_steps=0', '--max_train_stepsdation_prompt=A photo of sks dog in a bucket', '--validation_epochs=25', '--seed=0', '--push_to_hub']' returned non-zero exit status 1.
复制代码
注:上述文件注意存放在同一文件夹下。如果仍有题目可以在Github项目中的issue栏目举行搜索 or 提问 / 询问ChatGPT / 欣赏器搜索,Github上issue位置如下所示。