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

Resume from checkpoint #36

Open
janniehues opened this issue Nov 5, 2022 · 6 comments
Open

Resume from checkpoint #36

janniehues opened this issue Nov 5, 2022 · 6 comments

Comments

@janniehues
Copy link

Hello,

Thank you for giving this example on how to finetune stable diffusion!
The training seems to be working fine. However, when I try to resume from an intermediate checkpoint that was created upon training, the reconstructed images come out to be just noisy blurs. Any ideas what I am doing wrong here when resuming training from a checkpoint that was created on the way? I just changed the .cpkt in finetune_from and also tried the resume_from flag, but both did not work. Thank you for your help

@cxhermagic
Copy link

I have the same issue, please help, thank you .

@nocol0101001
Copy link

I also have the same problem, how to solve it. Thanks

@nocol0101001
Copy link

!(python main.py
-t
--base configs/stable-diffusion/pokemon.yaml
--gpus "0,"
--scale_lr False
--num_nodes 1
--check_val_every_n_epoch 10
--finetune_from "$ckpt_path"
data.params.batch_size="2"
lightning.trainer.accumulate_grad_batches="1"
data.params.validation.params.n_gpus="$NUM_GPUS"
)

Hello, what I get from training like this is still a mass of noise, what is the cause. Thanks

@janniehues
Copy link
Author

Hello,

To fine-tune stable diffusion I switched to the original repository now. You need to copy the bits from this repository that is in main.py and where the model state is read in. Then just resume training with the code from the original stable diffusion github.

@nocol0101001
Copy link

nocol0101001 commented May 4, 2023 via email

@janniehues
Copy link
Author

Ok 10 pictures are not very many. Maybe have a look at the dreambooth algorithm https://arxiv.org/abs/2208.12242 then. That should also have a github repo somewhere .

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

3 participants