diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
| commit | 339b5315700a469f4a9f0d5afc08ca2aca60c579 (patch) | |
| tree | 154d6ed405883ca47e23d67e6d3f7699ff4e23e3 /modules/sd_models.py | |
| parent | a6e653be26cc05f4438145fa0082816e9fbbf5fc (diff) | |
| download | stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.gz stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.bz2 stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.zip | |
custom unet support
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 91b3eb11..835bc016 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -14,7 +14,7 @@ import ldm.modules.midas as midas from ldm.util import instantiate_from_config
-from modules import paths, shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization, errors, hashes, sd_models_config
+from modules import paths, shared, modelloader, devices, script_callbacks, sd_vae, sd_disable_initialization, errors, hashes, sd_models_config, sd_unet
from modules.sd_hijack_inpainting import do_inpainting_hijack
from modules.timer import Timer
import tomesd
@@ -532,6 +532,8 @@ def reload_model_weights(sd_model=None, info=None): if sd_model.sd_model_checkpoint == checkpoint_info.filename:
return
+ sd_unet.apply_unet("None")
+
if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
lowvram.send_everything_to_cpu()
else:
|