replicate / dreambooth-template

A template repo for training and publishing your own custom Stable Diffusion model using https://replicate.com/replicate/dreambooth
Apache License 2.0
51 stars 10 forks source link

Add publish script, change train script #3

Closed evilstreak closed 2 years ago

evilstreak commented 2 years ago

I haven't tested the publish script, because my connection isn't up to it in any reasonable time for this evening. The training script seems to work, though I think we need to extract the weights to somewhere consistent so the publish script can find them.

zeke commented 2 years ago

Thanks for opening a PR to make this reviewable. 👍🏼

Let's ship it and iterate.