MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT-LLM / load_hf_tokenizer

Method load_hf_tokenizer

tensorrt_llm/llmapi/llm_utils.py:550–559  ·  view source on GitHub ↗
(model_dir,
                          trust_remote_code: bool = True,
                          use_fast: bool = True,
                          **kwargs)

Source from the content-addressed store, hash-verified

548
549 @staticmethod
550 def load_hf_tokenizer(model_dir,
551 trust_remote_code: bool = True,
552 use_fast: bool = True,
553 **kwargs) -> Optional[TransformersTokenizer]:
554 if (tokenizer := load_hf_tokenizer(model_dir, trust_remote_code,
555 use_fast, **kwargs)) is not None:
556 return tokenizer
557 else:
558 logger.warning(f"Failed to load tokenizer from {model_dir}")
559 return None
560
561 @staticmethod
562 def load_hf_generation_config(

Callers 3

_try_load_tokenizerMethod · 0.80
saveMethod · 0.80

Calls 2

load_hf_tokenizerFunction · 0.85
warningMethod · 0.45

Tested by

no test coverage detected