fix

huggingface · Sep 18, 2023 · 62b8974 · 62b8974
1 parent 8d102f7
commit 62b8974
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/onnx/test_onnx_graph_transformations.py b/tests/onnx/test_onnx_graph_transformations.py
@@ -85,6 +85,7 @@ def test_merge_decoders(self, *args):
                 tmpdir,
                 task=task,
                 no_post_process=True,
+                legacy=True,
             )
 
             decoder = onnx.load(os.path.join(tmpdir, "decoder_model.onnx"))