diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-07-08 12:10:10 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-08 12:10:10 +0000 |
| commit | ec9bbda3da846b4aa2f03b1e0f0952ffbc61f4f6 (patch) | |
| tree | 5ae676b6ac7242ad082ab7421cf8ae9d50d52818 /modules/lowvram.py | |
| parent | c4c63dd5e4760c56405cef2e71abc5c3604c4578 (diff) | |
| parent | 18256c5f0174126cb103afece2b39b6b831e034a (diff) | |
| download | stable-diffusion-webui-gfx803-ec9bbda3da846b4aa2f03b1e0f0952ffbc61f4f6.tar.gz stable-diffusion-webui-gfx803-ec9bbda3da846b4aa2f03b1e0f0952ffbc61f4f6.tar.bz2 stable-diffusion-webui-gfx803-ec9bbda3da846b4aa2f03b1e0f0952ffbc61f4f6.zip | |
Merge branch 'dev' into img2img-batch-png-info
Diffstat (limited to 'modules/lowvram.py')
| -rw-r--r-- | modules/lowvram.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/modules/lowvram.py b/modules/lowvram.py index e254cc13..d95bcfbf 100644 --- a/modules/lowvram.py +++ b/modules/lowvram.py @@ -15,6 +15,8 @@ def send_everything_to_cpu(): def setup_for_low_vram(sd_model, use_medvram):
+ sd_model.lowvram = True
+
parents = {}
def send_me_to_gpu(module, _):
@@ -96,3 +98,7 @@ def setup_for_low_vram(sd_model, use_medvram): diff_model.middle_block.register_forward_pre_hook(send_me_to_gpu)
for block in diff_model.output_blocks:
block.register_forward_pre_hook(send_me_to_gpu)
+
+
+def is_enabled(sd_model):
+ return getattr(sd_model, 'lowvram', False)
|