From 64a5de51f32834c1ae5395fc4d37a2df35681852 Mon Sep 17 00:00:00 2001 From: wanghailu Date: Mon, 27 Feb 2023 10:22:03 +0800 Subject: [PATCH] fix --- pyinfinitensor/src/pyinfinitensor/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index 4ec4f365..e9424ab3 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -588,7 +588,7 @@ class OnnxStub: self.handler.copy_float(self.inputs[name], data) def take_float(self) -> List[float]: - return next(iter(self.handler.outputs.values())).copyFloats() + return next(iter(self.outputs.values())).cloneFloats() def from_onnx(model: ModelProto, runtime):