Code used to fine-tune Stable Diffusion on wallpaper images. A project made during the Young Talent program at AI Sweden.
The complete guide that we followed: https://lambdalabs.com/blog/how-to-fine-tune-stable-diffusion-how-we-made-the-text-to-pokemon-model-at-lambda
The command we used to initiate training: python main.py -t --base configs/stable-diffusion/low-lr1-5.yaml --gpus 1 --scale_lr False --num_nodes 1 --check_val_every_n_epoch 5 --finetune_from trainModel/v1-5-pruned.ckpt
Example images:
