aboutsummaryrefslogtreecommitdiffstats
path: root/modules/lowvram.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-07-31 21:24:48 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2023-07-31 21:24:48 +0000
commitb235022c615a7384f73c05fe240d8f4a28d103d4 (patch)
treef1534c9e6b78dcc0c13fe2833c6ca3c8f6fcb98f /modules/lowvram.py
parent6f0abbb71a3f29d6df63fed82d5d5e196ca0d4de (diff)
downloadstable-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/lowvram.py')
-rw-r--r--modules/lowvram.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/lowvram.py b/modules/lowvram.py
index 3f830664..96f52b7b 100644
--- a/modules/lowvram.py
+++ b/modules/lowvram.py
@@ -15,6 +15,9 @@ def send_everything_to_cpu():
def setup_for_low_vram(sd_model, use_medvram):
+ if getattr(sd_model, 'lowvram', False):
+ return
+
sd_model.lowvram = True
parents = {}