Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
Hui-88 opened this issue Jul 11, 2024 · 1 comment
Open

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

Hui-88 opened this issue Jul 11, 2024 · 1 comment

Comments

@Hui-88
Copy link

Hui-88 commented Jul 11, 2024

No description provided.

@Hui-88
Copy link
Author

Hui-88 commented Jul 11, 2024

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…

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant