NeoAnd/Dolphin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
yingdong.han 2025-06-30 19:51:26 +08:00
parent 0705bc12ce
commit cab9b3f952
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deployment/tensorrt_llm/ReadMe.md
View File

@ -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.