Skip to content

Commit 51e5cdf

Browse files
committed
fix
1 parent 2ea8a94 commit 51e5cdf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

_unittests/ut_torch_export_patches/test_patch_transformers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -379,9 +379,11 @@ def _get_cu_seqlens(cls):
379379
@requires_transformers("4.55")
380380
@unittest.skipIf(not patch_qwen2_5, "Qwen25 not part of this transformers")
381381
def test_patched_qwen2_5_vl_vision_attention_forward(self):
382+
from onnx_diagnostic.torch_export_patches.patches.patch_helper import (
383+
_is_torchdynamo_exporting,
384+
)
382385
from onnx_diagnostic.torch_export_patches.patches.patch_transformers import (
383386
patched_Qwen2_5_VLVisionAttention,
384-
_is_torchdynamo_exporting,
385387
)
386388

387389
config = get_cached_configuration("Qwen/Qwen2.5-VL-7B-Instruct")

0 commit comments

Comments
 (0)