diff --git a/modules/models_settings.py b/modules/models_settings.py index ca87b44b..bc3ace6f 100644 --- a/modules/models_settings.py +++ b/modules/models_settings.py @@ -96,7 +96,7 @@ def update_model_parameters(state, initial=False): gpu_memories.append(value) continue - if initial and vars(shared.args)[element] != vars(shared.args_defaults)[element]: + if initial and element in shared.provided_arguments: continue # Setting null defaults diff --git a/modules/shared.py b/modules/shared.py index ab1dbe11..2c562d2b 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -1,4 +1,5 @@ import argparse +import sys from collections import OrderedDict from pathlib import Path @@ -6,7 +7,6 @@ import yaml from modules.logging_colors import logger - # Model variables model = None tokenizer = None @@ -187,6 +187,11 @@ parser.add_argument('--multimodal-pipeline', type=str, default=None, help='The m args = parser.parse_args() args_defaults = parser.parse_args([]) +provided_arguments = [] +for arg in sys.argv[1:]: + arg = arg.lstrip('-').replace('-', '_') + if hasattr(args, arg): + provided_arguments.append(arg) # Deprecation warnings for k in ['chat', 'notebook', 'no_stream']: