From daa87ec914a4da01682d8b22e737a4a4c5d4bae8 Mon Sep 17 00:00:00 2001 From: luc <59030697+jaxs-ribs@users.noreply.github.com> Date: Thu, 9 Nov 2023 15:32:51 +0100 Subject: [PATCH] Fix wrong `onnx.load_model` arguments (#188) We called this with a second argument that got ignored in onnx 1.14.1, but with 1.15.0, this is not legal anymore. Removing the second argument should fix this. --- emote/extra/onnx_exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/emote/extra/onnx_exporter.py b/emote/extra/onnx_exporter.py index 3bc48e06..843307d8 100644 --- a/emote/extra/onnx_exporter.py +++ b/emote/extra/onnx_exporter.py @@ -206,7 +206,7 @@ def _trace(self): ) f.seek(0) - model_proto = onnx.load_model(f, onnx.ModelProto) + model_proto = onnx.load_model(f) return model_proto