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

Update initialization.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-26 16:11:29 -08:00
parent 4a428f954b
commit b91ada8df3
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/initialization.py
View File

@@ -14,8 +14,12 @@ def initialize_forge():
print('User disabled VRAM offload.') print('User disabled VRAM offload.')
model_management.ALWAYS_VRAM_OFFLOAD = False model_management.ALWAYS_VRAM_OFFLOAD = False
elif model_management.total_vram > 20 * 1024: elif model_management.total_vram > 20 * 1024:
print('Automatically disable VRAM offload since user have more than 20GB VRAM.') if args_parser.args.always_offload_from_vram:
model_management.ALWAYS_VRAM_OFFLOAD = False print('User has more than 20GB VRAM, but forced offloading models from VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = True
else:
print('Automatically disable VRAM offload since user has more than 20GB VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = False
else: else:
print('Always offload models from VRAM.') print('Always offload models from VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = True model_management.ALWAYS_VRAM_OFFLOAD = True