File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed
_unittests/ut_torch_export_patches
onnx_diagnostic/torch_export_patches/patches Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff 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 (
Original file line number Diff line number Diff line change 7676 patched_Qwen2_5_VisionTransformerPretrainedModel ,
7777 patched_Qwen2_5_VLVisionAttentionOneIteration ,
7878 patched_Qwen2_5_VLVisionAttention ,
79+ PLUGS as PLUGS_Qwen25 ,
7980 )
8081
8182from ._patch_transformers_qwen3 import patch_qwen3
You can’t perform that action at this time.
0 commit comments