search

divamgupta / diffusionbee-stable-diffusion-ui

Diffusion Bee is the easiest way to run Stable Diffusion locally on your M1 Mac. Comes with a one-click installer. No dependencies or technical knowledge needed.
https://diffusionbee.com
GNU Affero General Public License v3.0
12.4k stars 608 forks source link

Feature Request: Allow For replacing the model with our own finetuned models? #89

Open Maki9009 opened 1 year ago

Maki9009 commented 1 year ago

when i replace the model with my finetune model, the application gives me an error message and then crashes.

duanemoody commented 1 year ago

The application currently uses the Keras .h5 model format rather than TensorFlow checkpoint .ckpt; supposedly the huggingface codebase can import ckpt but I'm not certain how that works: https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

Zimbakin commented 1 year ago

+1 on this request. Many, including myself have trained .ckpt models.

Would also be an improvement to have the option to store the model in a custom location. The model in a hidden folder is causing problems if people want to uninstall - bloatwear.

enzyme69 commented 1 year ago

I also +1 this, can we replace the model with Waifu Diffusion Model for example? Or our own style trained model? (not sure how this work).

myneodream commented 1 year ago

Yes, it would be great

milesfoxy commented 1 year ago

Is there a .ckpt to .h5 converter out there? Not as tech-savvy with source code stuff here.

thezakman commented 1 year ago

So there is a way to convert .h5 to .ckpt or is possible to add the option to load other models?