fix

Author: xadupre

onnx_diagnostic/export/onnx_plug.py

@@ -188,7 +188,6 @@ def __init__(
         self.kwargs_name = [p for p in params if p in self.kwargs]
         self.verbose = verbose
         self.custom_op = self._register()
-        assert "com.microsoft" in {d.domain for d in function_proto.opset_import}
 
     @property
     def domain(self) -> str:

onnx_diagnostic/torch_export_patches/patches/_patch_transformers_qwen2_5.py

@@ -44,8 +44,8 @@ def LoopMHAAttention(
         seq_axis = op.Constant(value_ints=[1])
         seq_axis_int32 = op.Cast(seq_axis, to=onnx.TensorProto.INT32)
         attn_output = op.Slice(value_3d, [0], [0], seq_axis)
-        for i in range(num_patches):
-            i_1d = op.Reshape(i, [1])
+        for i_patch in range(num_patches):
+            i_1d = op.Reshape(i_patch, [1])
             i_plus_1_1d = i_1d + 1
             start = op.Gather(cu_seqlens, i_1d, axis=0)
             end = op.Gather(cu_seqlens, i_plus_1_1d, axis=0)