search

NickLucche / stable-diffusion-nvidia-docker

GPU-ready Dockerfile to run Stability.AI stable-diffusion model v2 with a simple web interface. Includes multi-GPUs support.
MIT License
357 stars 43 forks source link

loading custom models #28

Open Anonym0us33 opened 1 year ago

Anonym0us33 commented 1 year ago

Is there any way to use this without a hugging face token, or to change the SD model to something besides 1.5? Or is it part of the docker script? Could you possibly edit the docker script so that it no longer requires a HF download?

NickLucche commented 1 year ago

Yes this is definitely doable but currently it requires you to set the -e MODEL_ID <model_name> environment variable before starting the container to load a different model. Currently the version on master defaults to stabilityai/stable-diffusion-2-base. I will look into making this more user-friendly.

Could you possibly edit the docker script so that it no longer requires a HF download? This can be done already by mounting hf cache directory, see here https://github.com/NickLucche/stable-diffusion-nvidia-docker/tree/master#about-model-versions.

luffyx10 commented 1 year ago

Yes this is definitely doable but currently it requires you to set the -e MODEL_ID <model_name> environment variable before starting the container to load a different model. Currently the version on master defaults to stabilityai/stable-diffusion-2-base. I will look into making this more user-friendly.

Could you possibly edit the docker script so that it no longer requires a HF download? This can be done already by mounting hf cache directory, see here https://github.com/NickLucche/stable-diffusion-nvidia-docker/tree/master#about-model-versions.

Hi, Thank you for this and I have a problem loading other models can you please make a detailed instruction because am really lost and how can I add safetensors checkpoints Thanks again