Skip to content

Commit ba991d5

Browse files
authored
[Bugfix] Fix FlexibleArgumentParser replaces _ with - for actual args (#5795)
1 parent 1744cc9 commit ba991d5

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

vllm/utils.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -822,7 +822,13 @@ def parse_args(self, args=None, namespace=None):
822822
processed_args = []
823823
for arg in args:
824824
if arg.startswith('--'):
825-
processed_args.append('--' + arg[len('--'):].replace('_', '-'))
825+
if '=' in arg:
826+
key, value = arg.split('=', 1)
827+
key = '--' + key[len('--'):].replace('_', '-')
828+
processed_args.append(f'{key}={value}')
829+
else:
830+
processed_args.append('--' +
831+
arg[len('--'):].replace('_', '-'))
826832
else:
827833
processed_args.append(arg)
828834

0 commit comments

Comments
 (0)