diff --git a/tensorrt_llm/_torch/models/modeling_utils.py b/tensorrt_llm/_torch/models/modeling_utils.py index 22d544f92b..c4e7906bbd 100755 --- a/tensorrt_llm/_torch/models/modeling_utils.py +++ b/tensorrt_llm/_torch/models/modeling_utils.py @@ -71,6 +71,7 @@ class MetaInitMode(TorchDispatchMode): random_init_ops = { aten.normal_.default, aten.uniform_.default, + aten.log.default, # TODO: this is not a exhaustive list for random init ops, add as needed }