Closed ZeroCool22 closed 1 year ago
Trying to train runwayml/stable-diffusion-inpainting using this script:
https://github.com/ShivamShrirao/diffusers/blob/main/examples/dreambooth/train_inpainting_dreambooth.py
My launch_inpaint.sh:
`export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH export MODEL_NAME="runwayml/stable-diffusion-inpainting" export INSTANCE_DIR="training" export OUTPUT_DIR="my_model"
accelerate launch train_inpainting_dreambooth.py \ --pretrained_vae_name_or_path="stabilityai/sd-vae-ft-mse" \ --pretrained_model_name_or_path=$MODEL_NAME \ --instance_data_dir=$INSTANCE_DIR \ --output_dir=$OUTPUT_DIR \ --instance_prompt="Greyalieninpa" \ --resolution=512 \ --train_batch_size=1 \ --learning_rate=1e-6 \ --lr_scheduler="constant" \ --lr_warmup_steps=0 \ --gradient_accumulation_steps=4 --gradient_checkpointing \ --use_8bit_adam \ --save_interval=500 \ --max_train_steps=5000`
Try to run it and get the error above mentioned.
No response
diffusers
Describe the bug
Trying to train runwayml/stable-diffusion-inpainting using this script:
https://github.com/ShivamShrirao/diffusers/blob/main/examples/dreambooth/train_inpainting_dreambooth.py
My launch_inpaint.sh:
`export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH export MODEL_NAME="runwayml/stable-diffusion-inpainting" export INSTANCE_DIR="training" export OUTPUT_DIR="my_model"
accelerate launch train_inpainting_dreambooth.py \ --pretrained_vae_name_or_path="stabilityai/sd-vae-ft-mse" \ --pretrained_model_name_or_path=$MODEL_NAME \ --instance_data_dir=$INSTANCE_DIR \ --output_dir=$OUTPUT_DIR \ --instance_prompt="Greyalieninpa" \ --resolution=512 \ --train_batch_size=1 \ --learning_rate=1e-6 \ --lr_scheduler="constant" \ --lr_warmup_steps=0 \ --gradient_accumulation_steps=4 --gradient_checkpointing \ --use_8bit_adam \ --save_interval=500 \ --max_train_steps=5000`
Reproduction
Try to run it and get the error above mentioned.
Logs
No response
System Info
diffusers
version: 0.9.0