diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-01 00:24:48 +0300 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-01 00:24:48 +0300 |
| commit | b235022c615a7384f73c05fe240d8f4a28d103d4 (patch) | |
| tree | f1534c9e6b78dcc0c13fe2833c6ca3c8f6fcb98f /modules/sd_hijack.py | |
| parent | 6f0abbb71a3f29d6df63fed82d5d5e196ca0d4de (diff) | |
option to keep multiple models in memory
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py index cfa5f0eb..7d692e3c 100644 --- a/modules/sd_hijack.py +++ b/modules/sd_hijack.py @@ -30,8 +30,10 @@ ldm.modules.attention.MemoryEfficientCrossAttention = ldm.modules.attention.Cros ldm.modules.attention.BasicTransformerBlock.ATTENTION_MODES["softmax-xformers"] = ldm.modules.attention.CrossAttention
# silence new console spam from SD2
-ldm.modules.attention.print = lambda *args: None
-ldm.modules.diffusionmodules.model.print = lambda *args: None
+ldm.modules.attention.print = shared.ldm_print
+ldm.modules.diffusionmodules.model.print = shared.ldm_print
+ldm.util.print = shared.ldm_print
+ldm.models.diffusion.ddpm.print = shared.ldm_print
optimizers = []
current_optimizer: sd_hijack_optimizations.SdOptimization = None
|