kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][chore] Print log with time for starting to load safetensor weights (#8218)

Browse Source 
Signed-off-by: Hui Gao <huig@nvidia.com>
This commit is contained in:
HuiGao-NV 2025-10-10 13:54:54 +08:00 committed by GitHub
parent e8c9bae37e
commit 795a051765
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorrt_llm/_torch/models/checkpoints/hf/weight_loader.py
View File

@ -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