Fix command-line arguments being ignored

This commit is contained in:
oobabooga 2023-09-19 13:11:46 -07:00
parent ff5d3d2d09
commit 5075087461
2 changed files with 7 additions and 2 deletions

View File

@ -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

View File

@ -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']: