how to run the unet.onnx model

#2
by Molten38 - opened

Hello, I have loaded the unet.onnx model as follows:

import onnx
from onnx.external_data_helper import load_external_data_for_model

onnx_model = onnx.load("unet\\unet.onnx", load_external_data=True)
load_external_data_for_model(onnx_model, "unet\\")

But when I try to run it via ort, I get the following error
onnxruntime.capi.onnxruntime_pybind11_state.InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOBUF

how can I properly load the unet.onnx model aswell as the corresponding weights to make it run in ort?

Hey Molten38 Did you ever find a way around this?

Sign up or log in to comment