epitaque / dreambooth_depth2img

adaptation of huggingface's dreambooth training script to support depth2img
MIT License
101 stars 15 forks source link

howto control model size #8

Open chris-aeviator opened 1 year ago

chris-aeviator commented 1 year ago

I realized when training the model with your script, my images output as 384 x 384 px . I have not yet used Diffusers directly for inference before, but from my understanding the model size for SD 2-0 depth is at least 512 px. How can I control this during training?