Fix kwarg name (#2691)

NVIDIA · Jan 20, 2025 · d93a2dd · d93a2dd
1 parent 0d0583a
commit d93a2dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: