Skip to content

Commit 0b6977f

Browse files
committed
fix: broken cast
Signed-off-by: ixlmar <[email protected]>
1 parent cd4e639 commit 0b6977f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorrt_llm/llmapi/llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -459,8 +459,8 @@ def generate_async(
459459
elif 'multi_modal_embeddings' in inputs:
460460
mm_embedding_info = inputs['multi_modal_embeddings']
461461
prompt_token_ids, extra_processed_inputs = cast(
462-
self.input_processor,
463-
BaseMultimodalInputProcessor).attach_multimodal_embeddings(
462+
BaseMultimodalInputProcessor,
463+
self.input_processor).attach_multimodal_embeddings(
464464
inputs, mm_embedding_info, sampling_params)
465465
else:
466466
with nvtx_range_debug("input_processor"):

0 commit comments

Comments
 (0)