From 83b56d28637317a5a14ef91808c6a3a3026cbfc7 Mon Sep 17 00:00:00 2001 From: Robert Knight Date: Wed, 22 May 2024 00:03:35 +0100 Subject: [PATCH] Fix conversion of `Transpose` operator with missing `perm` attr A missing `perm` attr signals that the input dimensions should be reversed. The converter translated a missing `perm` attr into an empty permuation list instead of `None`, causing the operator to fail with an "Invalid permutation" error. Tested with the `decoder_model.onnx` model from https://huggingface.co/Mozilla/distilvit/tree/main/onnx. --- rten-convert/rten_convert/converter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rten-convert/rten_convert/converter.py b/rten-convert/rten_convert/converter.py index 2d3bd76e..70beb6d0 100644 --- a/rten-convert/rten_convert/converter.py +++ b/rten-convert/rten_convert/converter.py @@ -921,7 +921,7 @@ def op_node_from_onnx_operator( case "Transpose": attrs = sg.TransposeAttrsT() - attrs.perm = op_reader.get_attr("perm", "ints", []) + attrs.perm = op_reader.get_attr("perm", "ints", None) case "Trilu": attrs = sg.TriluAttrsT()