From ba991d5c84adbc0685075af88333c688ddb06011 Mon Sep 17 00:00:00 2001 From: Chang Su Date: Mon, 24 Jun 2024 16:01:19 -0700 Subject: [PATCH] [Bugfix] Fix FlexibleArgumentParser replaces _ with - for actual args (#5795) --- vllm/utils.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/vllm/utils.py b/vllm/utils.py index ce5c377e..f0c7df5c 100644 --- a/vllm/utils.py +++ b/vllm/utils.py @@ -822,7 +822,13 @@ class FlexibleArgumentParser(argparse.ArgumentParser): processed_args = [] for arg in args: if arg.startswith('--'): - processed_args.append('--' + arg[len('--'):].replace('_', '-')) + if '=' in arg: + key, value = arg.split('=', 1) + key = '--' + key[len('--'):].replace('_', '-') + processed_args.append(f'{key}={value}') + else: + processed_args.append('--' + + arg[len('--'):].replace('_', '-')) else: processed_args.append(arg)