Zihao Wang

Hi @patrickvonplaten, I followed the instructions of https://github.com/huggingface/diffusers/tree/main/examples/dreambooth until the Dog Toy Example: ``` export MODEL_NAME="CompVis/stable-diffusion-v1-4" export INSTANCE_DIR="path-to-instance-images" export OUTPUT_DIR="path-to-save-model" accelerate launch train_dreambooth.py \ --pretrained_model_name_or_path=$MODEL_NAME \ --instance_data_dir=$INSTANCE_DIR \ --output_dir=$OUTPUT_DIR \...

Hi @pcuenca! Thanks for your reply! * The training stalls at this line `accelerator.backward(loss)` https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/train_dreambooth.py#L675 * I am using a single 40 GB A100 GPU. Output from `diffusers-cli env`: ```...