From cb34e10956b5a07a6c71bb84f1036c09ac9b4ca7 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 29 Jan 2024 22:52:11 -0800 Subject: [PATCH] Update initialization.py --- modules_forge/initialization.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/modules_forge/initialization.py b/modules_forge/initialization.py index 32c67fe7..6c3839e7 100644 --- a/modules_forge/initialization.py +++ b/modules_forge/initialization.py @@ -1,5 +1,18 @@ +import sys + def initialize_forge(): + bad_list = ['--lowvram', '--medvram', '--medvram-sdxl'] + + for bad in bad_list: + if bad in sys.argv: + print(f'Arg {bad} is removed in Forge.') + print(f'Now memory management is fully automatic and you do not need any command flags.') + print(f'Please just remove this flag.') + print(f'In extreme cases, if you want to force previous lowvram/medvram behaviors, ' + f'please use --always-offload-from-vram') + exit(0) + from ldm_patched.modules import args_parser args_parser.args, _ = args_parser.parser.parse_known_args()