diff --git a/vllm/entrypoints/llm.py b/vllm/entrypoints/llm.py index 7297243f9188..892e5035ab60 100644 --- a/vllm/entrypoints/llm.py +++ b/vllm/entrypoints/llm.py @@ -556,7 +556,8 @@ def wait_for_completion( and returns their outputs. Use after enqueue() to get results. Args: - output_type: The expected output type, defaults to RequestOutput. + output_type: The expected output type(s). If not provided, accepts + both RequestOutput and PoolingRequestOutput. use_tqdm: If True, shows a tqdm progress bar. Returns: