Skip to content

Commit fbdcb87

Browse files
committed
fix serialization
Signed-off-by: Superjomn <[email protected]>
1 parent d516d02 commit fbdcb87

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tensorrt_llm/executor/serialization.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@
8686
"tensorrt_llm.llmapi.mpi_session": ["RemoteTask"],
8787
"tensorrt_llm.llmapi.llm_utils":
8888
["CachedModelLoader._node_build_task", "LlmBuildStats"],
89+
"tensorrt_llm.llmapi.tokenizer": ["TransformersTokenizer"],
8990
"tensorrt_llm.lora_manager": ["LoraConfig"],
9091
"tensorrt_llm.mapping": ["Mapping"],
9192
"tensorrt_llm.models.modeling_utils":

0 commit comments

Comments
 (0)