I want to change onnx model and then restore the weights when using it with onnxruntime. However, the m