Skip to content

Commit 87fe44f

Browse files
authored
feat(models): Mistral3.1 VLM pytorch backend support (#5529)
Signed-off-by: William Zhang <[email protected]>
1 parent b61a717 commit 87fe44f

File tree

8 files changed

+1224
-4
lines changed

8 files changed

+1224
-4
lines changed

pyproject.toml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ extend_skip_glob = [
1818
"tensorrt_llm/_dlpack_utils.py",
1919
"tensorrt_llm/_ipc_utils.py",
2020
"tensorrt_llm/_mnnvl_utils.py",
21+
"tensorrt_llm/_torch/models/modeling_pixtral.py",
2122
"tensorrt_llm/disaggregated_params.py",
2223
"tensorrt_llm/engine.py",
2324
"tensorrt_llm/graph_rewriting.py",
@@ -30,6 +31,8 @@ extend_skip_glob = [
3031
"tensorrt_llm/python_plugin.py",
3132
"tensorrt_llm/sampling_params.py",
3233
"tensorrt_llm/top_model_mixin.py",
34+
"tests/unittest/_torch/modeling/test_modeling_mistral.py",
35+
"tests/unittest/_torch/modeling/test_modeling_pixtral.py",
3336
]
3437

3538
[tool.yapf]
@@ -45,6 +48,7 @@ ignore_patterns = [
4548
"tensorrt_llm/_dlpack_utils.py",
4649
"tensorrt_llm/_ipc_utils.py",
4750
"tensorrt_llm/_mnnvl_utils.py",
51+
"tensorrt_llm/_torch/models/modeling_pixtral.py",
4852
"tensorrt_llm/disaggregated_params.py",
4953
"tensorrt_llm/engine.py",
5054
"tensorrt_llm/graph_rewriting.py",
@@ -57,6 +61,8 @@ ignore_patterns = [
5761
"tensorrt_llm/python_plugin.py",
5862
"tensorrt_llm/sampling_params.py",
5963
"tensorrt_llm/top_model_mixin.py",
64+
"tests/unittest/_torch/modeling/test_modeling_mistral.py",
65+
"tests/unittest/_torch/modeling/test_modeling_pixtral.py",
6066
]
6167

6268
[tool.codespell]
@@ -76,6 +82,7 @@ exclude = [
7682
"tensorrt_llm/_dlpack_utils.py",
7783
"tensorrt_llm/_ipc_utils.py",
7884
"tensorrt_llm/_mnnvl_utils.py",
85+
"tensorrt_llm/_torch/models/modeling_pixtral.py",
7986
"tensorrt_llm/disaggregated_params.py",
8087
"tensorrt_llm/engine.py",
8188
"tensorrt_llm/graph_rewriting.py",
@@ -88,6 +95,8 @@ exclude = [
8895
"tensorrt_llm/python_plugin.py",
8996
"tensorrt_llm/sampling_params.py",
9097
"tensorrt_llm/top_model_mixin.py",
98+
"tests/unittest/_torch/modeling/test_modeling_mistral.py",
99+
"tests/unittest/_torch/modeling/test_modeling_pixtral.py",
91100
]
92101

93102

@@ -116,6 +125,7 @@ include = [
116125
"tensorrt_llm/_dlpack_utils.py",
117126
"tensorrt_llm/_ipc_utils.py",
118127
"tensorrt_llm/_mnnvl_utils.py",
128+
"tensorrt_llm/_torch/models/modeling_pixtral.py",
119129
"tensorrt_llm/disaggregated_params.py",
120130
"tensorrt_llm/engine.py",
121131
"tensorrt_llm/graph_rewriting.py",
@@ -128,6 +138,8 @@ include = [
128138
"tensorrt_llm/python_plugin.py",
129139
"tensorrt_llm/sampling_params.py",
130140
"tensorrt_llm/top_model_mixin.py",
141+
"tests/unittest/_torch/modeling/test_modeling_mistral.py",
142+
"tests/unittest/_torch/modeling/test_modeling_pixtral.py",
131143
]
132144
exclude = [
133145
"**3rdparty/**",

0 commit comments

Comments
 (0)