[Bugfix] Fix FlexibleArgumentParser replaces _ with - for actual args (…

…vllm-project#5795)
zifeitong · Jun 24, 2024 · ba991d5 · ba991d5
1 parent 1744cc9
commit ba991d5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/vllm/utils.py b/vllm/utils.py
@@ -822,7 +822,13 @@ def parse_args(self, args=None, namespace=None):
         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)