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.
EmbarkStudios · Nov 9, 2023 · daa87ec · daa87ec
1 parent 36a1c3d
commit daa87ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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