Skip to content

Commit cb38d0b

Browse files
change to ValueError
Signed-off-by: yechank <[email protected]>
1 parent 39d522f commit cb38d0b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorrt_llm/_torch/models/modeling_qwen2vl.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -919,7 +919,8 @@ def __init__(
919919
self.model_config = model_config
920920
self.config = model_config.pretrained_config
921921

922-
assert model_config.attn_backend == 'TRTLLM', "Qwen2/2.5-VL only supports TRTLLM backend now"
922+
if model_config.attn_backend != 'TRTLLM':
923+
raise ValueError("Qwen2/2.5-VL only supports TRTLLM backend now")
923924
if not disabble_fuse_rope:
924925
self.init_mrope_embedding(model_config)
925926

0 commit comments

Comments
 (0)