[Bugfix] Fix FlexibleArgumentParser replaces _ with - for actual args (#5795)

ba991d5c · Chang Su · GitHub · 1744cc99 · ba991d5c
Unverified Commit ba991d5c authored Jun 24, 2024 by Chang Su Committed by GitHub Jun 24, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

vllm/utils.py vllm/utils.py +7 -1

No files found.
--- 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)