How to include a custom model? about stable-diffusion-webui-docker

[AbdBarho]

Fixed in #75

[sheldonhull]

I'm working through it with adding to downloads, and removing my download taskfile command since you've built this into the compose process. Lots of interesting different approaches here that I think are useful to give me new ideas!

I'll see if this works and update for others.

[AbdBarho]

@sheldonhull hlky UI? I don't think you can add custom models to it, I think that works for ESRGAN in auto only.
Can you link to documentation / description of the feature you are trying to use?

[sheldonhull]

I'm not super versed in this, learning the ecosystem. That's fine if I can't add anything, but was trying to experiment and see if I could mount a new model to show up in the UI with a ckpt file.
I was experimenting with waifu-difussion to see what img2img results I could obtain.

If it's not possible cool!

[AbdBarho]

@sheldonhull Ah, I see, waifu-difussion is just a fine tuned-model.

In theory, you don't need to change any code, you can just

mv cache/models/model.ckpt cache/models/model.backup.ckpt 
mv waifu-diffusion.ckpt cache/models/model.ckpt

and run the UI, any UI should work because this is just the weights.

[sheldonhull]

So this replaces the model.ckpt that's there, meaning it's not a switch for me, it's one or the other?

I plan on learning more on the technical side of this later, so sorry if this is obvious stuff. I'm only dabbling in this area to learn a bit and have some fun ;-)

[AbdBarho]

@sheldonhull correct, it is one or the other. if you want a switch through an env variable, update the CMD of services/hlky/Dockerfile and delete the --ckpt /cache/models/model.ckpt

and then you can define the model you want in the cli args of the ui

- CLI_ARGS=--optimized-turbo --ckpt /cache/models/waifu-diffusion.ckpt

You are going to have to restart the container for that to work.

If you want to update the model in runtime while the container is running, then you should probably create a feature request in the respective ui repo.