9
9
if TYPE_CHECKING :
10
10
from vllm .multimodal import (MultiModalDataDict , MultiModalKwargs ,
11
11
MultiModalPlaceholderDict )
12
- from vllm .multimodal .inputs import MultiModalInputsV2
12
+ from vllm .multimodal .inputs import MultiModalInputs
13
13
14
14
15
15
class TextPrompt (TypedDict ):
@@ -207,7 +207,7 @@ def token_inputs(
207
207
return inputs
208
208
209
209
210
- DecoderOnlyInputs = Union [TokenInputs , "MultiModalInputsV2 " ]
210
+ DecoderOnlyInputs = Union [TokenInputs , "MultiModalInputs " ]
211
211
"""
212
212
The inputs in :class:`~vllm.LLMEngine` before they are
213
213
passed to the model executor.
@@ -222,14 +222,14 @@ class EncoderDecoderInputs(TypedDict):
222
222
223
223
This specifies the required data for encoder-decoder models.
224
224
"""
225
- encoder : Union [TokenInputs , "MultiModalInputsV2 " ]
225
+ encoder : Union [TokenInputs , "MultiModalInputs " ]
226
226
"""The inputs for the encoder portion."""
227
227
228
- decoder : Union [TokenInputs , "MultiModalInputsV2 " ]
228
+ decoder : Union [TokenInputs , "MultiModalInputs " ]
229
229
"""The inputs for the decoder portion."""
230
230
231
231
232
- SingletonInputs = Union [TokenInputs , "MultiModalInputsV2 " ]
232
+ SingletonInputs = Union [TokenInputs , "MultiModalInputs " ]
233
233
"""
234
234
A processed :class:`SingletonPrompt` which can be passed to
235
235
:class:`vllm.sequence.Sequence`.
@@ -311,7 +311,7 @@ def multi_modal_hashes(self) -> List[str]:
311
311
return inputs .get ("multi_modal_hashes" , [])
312
312
313
313
if inputs ["type" ] == "multimodal" :
314
- # only the case when we use MultiModalInputsV2
314
+ # only the case when we use MultiModalInputs
315
315
return inputs .get ("mm_hashes" , []) # type: ignore[return-value]
316
316
317
317
assert_never (inputs ) # type: ignore[arg-type]
0 commit comments