diff --git a/fastdeploy/engine/args_utils.py b/fastdeploy/engine/args_utils.py index 9112efa7d..82a6ddbd2 100644 --- a/fastdeploy/engine/args_utils.py +++ b/fastdeploy/engine/args_utils.py @@ -669,7 +669,7 @@ class EngineArgs: model_group.add_argument( "--logprobs-mode", type=str, - choices=["raw_logprobs", "processed_logprobs", "processed_logits"], + choices=["raw_logprobs", "raw_logits", "processed_logprobs", "processed_logits"], default=EngineArgs.logprobs_mode, help="Indicates the content returned in the logprobs.", )