From b91ada8df37a856b81a88f36c9ef7a3285b04560 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 26 Jan 2024 16:11:29 -0800 Subject: [PATCH] Update initialization.py --- modules_forge/initialization.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/modules_forge/initialization.py b/modules_forge/initialization.py index 7d475a26..5d579ce1 100644 --- a/modules_forge/initialization.py +++ b/modules_forge/initialization.py @@ -14,8 +14,12 @@ def initialize_forge(): 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 + if args_parser.args.always_offload_from_vram: + 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: print('Always offload models from VRAM.') model_management.ALWAYS_VRAM_OFFLOAD = True