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

…vllm-project#5795)
opendatahub-io · Jun 28, 2024 · 2d30ec8 · 2d30ec8
1 parent b8897c0
commit 2d30ec8
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)