diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-31 21:24:48 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-31 21:24:48 +0000 |
| commit | b235022c615a7384f73c05fe240d8f4a28d103d4 (patch) | |
| tree | f1534c9e6b78dcc0c13fe2833c6ca3c8f6fcb98f /modules/sd_models_xl.py | |
| parent | 6f0abbb71a3f29d6df63fed82d5d5e196ca0d4de (diff) | |
| download | stable-diffusion-webui-gfx803-b235022c615a7384f73c05fe240d8f4a28d103d4.tar.gz stable-diffusion-webui-gfx803-b235022c615a7384f73c05fe240d8f4a28d103d4.tar.bz2 stable-diffusion-webui-gfx803-b235022c615a7384f73c05fe240d8f4a28d103d4.zip | |
option to keep multiple models in memory
Diffstat (limited to 'modules/sd_models_xl.py')
| -rw-r--r-- | modules/sd_models_xl.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/modules/sd_models_xl.py b/modules/sd_models_xl.py index bc219508..01123321 100644 --- a/modules/sd_models_xl.py +++ b/modules/sd_models_xl.py @@ -98,10 +98,10 @@ def extend_sdxl(model): model.conditioner.wrapped = torch.nn.Module()
-sgm.modules.attention.print = lambda *args: None
-sgm.modules.diffusionmodules.model.print = lambda *args: None
-sgm.modules.diffusionmodules.openaimodel.print = lambda *args: None
-sgm.modules.encoders.modules.print = lambda *args: None
+sgm.modules.attention.print = shared.ldm_print
+sgm.modules.diffusionmodules.model.print = shared.ldm_print
+sgm.modules.diffusionmodules.openaimodel.print = shared.ldm_print
+sgm.modules.encoders.modules.print = shared.ldm_print
# this gets the code to load the vanilla attention that we override
sgm.modules.attention.SDP_IS_AVAILABLE = True
|