[Frontend][Bugfix] Fix preemption_mode -> preemption-mode for CLI arg in arg_utils.py (#5688)

afed90a0 · Michael Goin · GitHub · 3ee5c4bc · afed90a0
Unverified Commit afed90a0 authored Jun 19, 2024 by Michael Goin Committed by GitHub Jun 19, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

vllm/engine/arg_utils.py vllm/engine/arg_utils.py +1 -1

No files found.
--- a/vllm/engine/arg_utils.py
+++ b/vllm/engine/arg_utils.py
@@ -577,7 +577,7 @@ class EngineArgs:
                            '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 '