From 4fe5c9fd559466526bd8869c9c24327d56063009 Mon Sep 17 00:00:00 2001 From: Denis Kayshev Date: Tue, 14 Jan 2025 13:21:31 +0300 Subject: [PATCH] Fix kwarg name --- tensorrt_llm/llmapi/llm_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorrt_llm/llmapi/llm_utils.py b/tensorrt_llm/llmapi/llm_utils.py index 7b9e0957665b..7b9b49725b8c 100644 --- a/tensorrt_llm/llmapi/llm_utils.py +++ b/tensorrt_llm/llmapi/llm_utils.py @@ -485,7 +485,7 @@ def __post_init__(self): else: self.tokenizer = tokenizer_factory( self.tokenizer, - rust_remote_code=self.trust_remote_code, + trust_remote_code=self.trust_remote_code, use_fast=self.tokenizer_mode != 'slow') if torch.cuda.get_device_properties(0).major < 8: