Got it to work ! Turns out the issue was from the decoder model. I exported it myself to an onnx model and now it works just fine !
