TheLastBen / fast-stable-diffusion

fast-stable-diffusion + DreamBooth
MIT License
7.49k stars 1.3k forks source link

[Feature request]dreambooth add other model support #279

Open OolongFox opened 1 year ago

OolongFox commented 1 year ago

I would like dreambooth to support other models, such as NovelAILeaks, but I'm a programming noob and don't know how to do that. Here's the link to the NAI model I found: https://pub-2fdef7a2969f43289c42ac5ae3412fd4.r2.dev/animefull-pruned.tar

TheLastBen commented 1 year ago

NovelAILeaks is an illegal asset

OolongFox commented 1 year ago

NovelAILeaks is an illegal asset

Ok, so can the waifu-diffusion model be supported please, here is the link to it:https://huggingface.co/hakurei/waifu-diffusion-v1-3.

TheLastBen commented 1 year ago

fixed, refresh your colab and check the box "Compatibility_mode" before downloading the model

shinzopro commented 1 year ago

I would like the ability to change the learning rate of Fast Dream Booth this Colab is very easy to use, but I thought this function was necessary to improve the variation in the loss values ​​being learned.

TheLastBen commented 1 year ago

You can do it easily by changing the code in the cell, just click twice on the cell and it will show the code.

shinzopro commented 1 year ago

thankyou

askiiart commented 1 year ago

fixed, refresh your colab and check the box "Compatibility_mode" before downloading the model

I believe this can be closed, then?