Open athenawisdoms opened 1 year ago
That is not my notebook, this one is https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast-DreamBooth.ipynb
OH I'm so sorry! I pasted the wrong link, updated it to be https://github.com/TheLastBen/diffusers/blob/main/examples/dreambooth/train_dreambooth.py
Is there an example to perform the training without using the ipynb notebook? Wondering why my command above doesnt produce a checkpoint file after training is done.
Hello there!
I'm using TheLastBen version of Dreambooth to train on a set of 10 images on a RTX 3090 24GB on Ubuntu with nvidia driver 525.105.17 and CUDA 12.0.
Unforunately, the final checkpoints cannot be found in the output directory, which contains only tensorboard log files.
Here's the output from the start till the end of the training:
How do I find the final checkpoint files to do inference on? Thanks!