diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-05 05:12:37 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-08-05 05:12:37 +0000 |
| commit | d56a9cfe6a0611465d01e6881f79c2d9128744de (patch) | |
| tree | 05878010fb1b638a0c6454fcff197fb0df3d4b93 /modules/sd_models_xl.py | |
| parent | a6b245e46f28efe013637e5e9b0600b88df79dc9 (diff) | |
| parent | a32f270a4758597ff97dfc55894e3cacf118ce5c (diff) | |
| download | stable-diffusion-webui-gfx803-d56a9cfe6a0611465d01e6881f79c2d9128744de.tar.gz stable-diffusion-webui-gfx803-d56a9cfe6a0611465d01e6881f79c2d9128744de.tar.bz2 stable-diffusion-webui-gfx803-d56a9cfe6a0611465d01e6881f79c2d9128744de.zip | |
Merge branch 'dev' into efficient-vae-methods
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
|