mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-25 02:15:01 +08:00
[test utils] correct wrong typing (#26159)
Signed-off-by: Yannick Schnider <yannick.schnider1@ibm.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
parent
c81dc099a3
commit
c634415273
@ -756,7 +756,7 @@ class VllmRunner:
|
||||
|
||||
def get_inputs(
|
||||
self,
|
||||
prompts: Union[list[str], list[torch.Tensor], list[int]],
|
||||
prompts: Union[list[str], list[torch.Tensor], list[list[int]]],
|
||||
images: Optional[PromptImageInput] = None,
|
||||
videos: Optional[PromptVideoInput] = None,
|
||||
audios: Optional[PromptAudioInput] = None,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user