[Frontend][Bugfix] Fix preemption_mode -> preemption-mode for CLI arg…

… in arg_utils.py (vllm-project#5688)
xjpang · Jul 24, 2024 · 8d9ee55 · 8d9ee55
1 parent 326cb11
commit 8d9ee55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/engine/arg_utils.py b/vllm/engine/arg_utils.py
@@ -577,7 +577,7 @@ def add_cli_args(
                             'This should be a JSON string that will be '
                             'parsed into a dictionary.')
         parser.add_argument(
-            '--preemption_mode',
+            '--preemption-mode',
             type=str,
             default=None,
             help='If \'recompute\', the engine performs preemption by block '