justinpinkney / stable-diffusion

MIT License
1.45k stars 266 forks source link

[feat] Fine tuning Stable Diffusion on Unsplash dataset #27

Closed blefaudeux closed 2 years ago

blefaudeux commented 2 years ago

python3 main.py -t --base configs/stable-diffusion/fine_tune_unsplash.yaml --gpus 2,3 --scale_lr False --num_nodes 1 --check_val_every_n_epoch 1 --finetune_from ../stable-diffusion-v-1-4-original/sd-v1-4-full-ema.ckpt