Skip to content

Commit 40b7b23

Browse files
committed
fix
1 parent 7c78810 commit 40b7b23

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

_unittests/ut_torch_export_patches/test_patch_transformers.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -384,6 +384,7 @@ def test_patched_qwen2_5_vl_vision_attention_forward(self):
384384
)
385385
from onnx_diagnostic.torch_export_patches.patches.patch_transformers import (
386386
patched_Qwen2_5_VLVisionAttention,
387+
PLUGS_Qwen25,
387388
)
388389

389390
config = get_cached_configuration("Qwen/Qwen2.5-VL-7B-Instruct")
@@ -456,6 +457,8 @@ def forward(
456457
dynamic_shapes=ds,
457458
exporter=exporter,
458459
filename=filename,
460+
onnx_plugs=PLUGS_Qwen25,
461+
target_opset=22,
459462
)
460463
# exporter_kwargs={"report":True} if exporter != "custom" else {}
461464
self.assert_onnx_disc(

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@
7676
patched_Qwen2_5_VisionTransformerPretrainedModel,
7777
patched_Qwen2_5_VLVisionAttentionOneIteration,
7878
patched_Qwen2_5_VLVisionAttention,
79+
PLUGS as PLUGS_Qwen25,
7980
)
8081

8182
from ._patch_transformers_qwen3 import patch_qwen3

0 commit comments

Comments
 (0)