Stability-AI / stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models
MIT License
39.25k stars 5.06k forks source link

What is the script for fine-tuning on one's own dataset? #388

Open Hui-88 opened 4 months ago

Hui-88 commented 4 months ago

python main.py \ -t \ --base configs/stable-diffusion/pokemon.yaml \ --gpus 0,1 \ --scale_lr False \ --num_nodes 1 \ --check_val_every_n_epoch 10 \ --finetune_from sd-v1-4-full-ema.ckpt I found such a script, but I'm not sure if that's right or not. Through this script, I fine tuned SD on my own 1000 datasets and trained 200 epochs, but the image generation effect was not good. 00526 ![Uploading 00606.png…]()