diff --git a/tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py b/tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py index d6660c95a7..7a631f45f6 100644 --- a/tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py +++ b/tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py @@ -92,6 +92,7 @@ class HfWeightLoader(BaseWeightLoader): @staticmethod def _load_safetensors_file(file): + logger.info(f"Start to load safetensor file {file}") return safetensors.torch.load_file(file) @staticmethod