search

TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.53k stars 1.31k forks source link

Static path to model #63

Closed GuruVirus closed 2 years ago

GuruVirus commented 2 years ago

Can you please add a field to reference a model's path locally in GoogleDrive instead of re-downloading thru HuggingFace every time (such a time waste and network waste)? I have like 3 copies of the same model because everyone's colab requires theirs to be in a different folder. I have no understanding of how multiple models are loaded so maybe add multiple paths or just a whole folder to be read now that there's custom models, waifu & robo diffusion out there.

This applies to all colabs (e.g. I've used your fast Automatic webUI and Dreambooth colabs).

TheLastBen commented 2 years ago

the dreambooth colab exports the ckpt to the main gdrive folder, then you can use the path of that ckpt and input it in AUTOMATIC1111 colab cell "download model".

GuruVirus commented 2 years ago

Apologies, it looks like you updated fast webUI colab a bunch since I last used it: https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast_stable_diffusion_AUTOMATIC1111.ipynb

I see you have a default path of /content/gdrive/MyDrive/sd/stable-diffusion-webui/models/Stable-diffusion/model.ckpt So as long as I have that in there it should skip the download, right?

And awesome to see "Path_to_trained_model"

TheLastBen commented 2 years ago

yes, anything existing on your Gdrive won't be redownloaded