[Frontend] Add --logprobs argument to benchmark_serving.py (#8191)

This commit is contained in:
afeldman-nm
2024-09-06 12:01:14 -04:00
committed by GitHub
parent baa5467547
commit e5cab71531
3 changed files with 19 additions and 1 deletions

View File

@@ -24,6 +24,7 @@ class RequestFuncInput:
model: str
best_of: int = 1
use_beam_search: bool = False
logprobs: Optional[int] = None
@dataclass
@@ -236,6 +237,7 @@ async def async_request_openai_completions(
"temperature": 0.0,
"best_of": request_func_input.best_of,
"max_tokens": request_func_input.output_len,
"logprobs": request_func_input.logprobs,
"stream": True,
}
headers = {