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

[None][fix] Properly raise error for nemotron H models (#8697)

Browse Source 
Signed-off-by: William Zhang <133824995+2ez4bz@users.noreply.github.com>
This commit is contained in:
William Zhang 2025-10-28 08:59:42 -07:00 committed by GitHub
parent 5a01f382c1
commit cdc9e5e645
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorrt_llm/_torch/models/modeling_nemotron_h.py
View File

@ -161,7 +161,7 @@ class NemotronHLayer(DecoderLayer):
elif layer_type == "*":
self.mixer = TransformerLayer(model_config, layer_idx)
else:
ValueError(f"{layer_type} is not supported")
raise ValueError(f"{layer_type} is not supported")
def forward(
self,