From 795a051765ac3c616902d894cf02f4a0c3aa5646 Mon Sep 17 00:00:00 2001 From: HuiGao-NV Date: Fri, 10 Oct 2025 13:54:54 +0800 Subject: [PATCH] [None][chore] Print log with time for starting to load safetensor weights (#8218) Signed-off-by: Hui Gao --- tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py | 1 + 1 file changed, 1 insertion(+) 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