lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-26 19:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

a bit better logic

Browse Source
This commit is contained in:
lllyasviel
2024-01-26 15:48:39 -08:00
parent 31a1ee761c
commit 4a428f954b
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
modules_forge/initialization.py
View File

@@ -7,9 +7,19 @@ def initialize_forge():
"Use this when you ara on MAC or have more than 20GB VRAM like RTX4096.")
args_parser.args = args_parser.parser.parse_known_args()[0]
args_parser.args.always_offload_from_vram = not args_parser.args.disable_offload_from_vram
import ldm_patched.modules.model_management as model_management
if args_parser.args.disable_offload_from_vram:
print('User disabled VRAM offload.')
model_management.ALWAYS_VRAM_OFFLOAD = False
elif model_management.total_vram > 20 * 1024:
print('Automatically disable VRAM offload since user have more than 20GB VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = False
else:
print('Always offload models from VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = True
import torch
device = model_management.get_torch_device()