Skip to content

Commit cdc9e5e

Browse files
authored
[None][fix] Properly raise error for nemotron H models (#8697)
Signed-off-by: William Zhang <[email protected]>
1 parent 5a01f38 commit cdc9e5e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorrt_llm/_torch/models/modeling_nemotron_h.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def __init__(
161161
elif layer_type == "*":
162162
self.mixer = TransformerLayer(model_config, layer_idx)
163163
else:
164-
ValueError(f"{layer_type} is not supported")
164+
raise ValueError(f"{layer_type} is not supported")
165165

166166
def forward(
167167
self,

0 commit comments

Comments
 (0)