mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-05 21:25:57 +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(
|
def get_inputs(
|
||||||
self,
|
self,
|
||||||
prompts: Union[list[str], list[torch.Tensor], list[int]],
|
prompts: Union[list[str], list[torch.Tensor], list[list[int]]],
|
||||||
images: Optional[PromptImageInput] = None,
|
images: Optional[PromptImageInput] = None,
|
||||||
videos: Optional[PromptVideoInput] = None,
|
videos: Optional[PromptVideoInput] = None,
|
||||||
audios: Optional[PromptAudioInput] = None,
|
audios: Optional[PromptAudioInput] = None,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user