diff options
| author | d8ahazard <d8ahazard@gmail.com> | 2022-09-27 16:01:13 +0000 |
|---|---|---|
| committer | d8ahazard <d8ahazard@gmail.com> | 2022-09-27 16:01:13 +0000 |
| commit | 11875f586323cea7c5b8398976449788a83dee76 (patch) | |
| tree | 49d57f8b5894ce5fd953ebf0b7c39ee927ca94e5 /webui.py | |
| parent | f860f94c64d98ff575132396b0636cc3e08f8332 (diff) | |
| download | stable-diffusion-webui-gfx803-11875f586323cea7c5b8398976449788a83dee76.tar.gz stable-diffusion-webui-gfx803-11875f586323cea7c5b8398976449788a83dee76.tar.bz2 stable-diffusion-webui-gfx803-11875f586323cea7c5b8398976449788a83dee76.zip | |
Use model loader with stable-diffusion too.
Hook the model loader into the SD_models file.
Add default url/download if checkpoint is not found.
Add matching stablediffusion-models-path argument.
Add message that --ckpt-dir will be removed in the future, but have it pipe to stablediffusion-models-path for now.
Update help strings for models-path args so they're more or less uniform.
Move sd_model "setup" call to webUI with the others.
Ensure "cleanup_models" method moves existing models to the new locations, including SD, and that we aren't deleting folders that still have stuff in them.
Diffstat (limited to 'webui.py')
| -rw-r--r-- | webui.py | 1 |
1 files changed, 1 insertions, 0 deletions
@@ -23,6 +23,7 @@ from modules.paths import script_path from modules.shared import cmd_opts
modelloader.cleanup_models()
+modules.sd_models.setup_model(cmd_opts.stablediffusion_models_path)
codeformer.setup_model(cmd_opts.codeformer_models_path)
gfpgan.setup_model(cmd_opts.gfpgan_models_path)
shared.face_restorers.append(modules.face_restoration.FaceRestoration())
|