From cab9b3f95284e44a9bc40184353707a6835d0b95 Mon Sep 17 00:00:00 2001 From: "yingdong.han" Date: Mon, 30 Jun 2025 19:51:26 +0800 Subject: [PATCH] update --- deployment/tensorrt_llm/ReadMe.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deployment/tensorrt_llm/ReadMe.md b/deployment/tensorrt_llm/ReadMe.md index ffc8c42..eb51810 100644 --- a/deployment/tensorrt_llm/ReadMe.md +++ b/deployment/tensorrt_llm/ReadMe.md @@ -7,7 +7,7 @@ The Dolphin model employs a **Swin Encoder + MBart Decoder** architecture. In th its architectures field is specified as "VisionEncoderDecoderModel". Dolphin, Nougat, and Donut share the same model architecture. TensorRT-LLM has already supported the Nougat model. Following Nougat's conversion script, we have successfully implemented Dolphin on TensorRT-LLM. -**Note:** prompt_ids MUST be of **int32** type, otherwise TensorRT-LLM will produce incorrect results. +**Note:** [prompt_ids](./dolphin_runner.py#L120) MUST be of **int32** type, otherwise TensorRT-LLM will produce incorrect results. ## 🛠️ Installation > We only test TensorRT-LLM 0.18.1 on Linux.