mithril-security / blindai

Confidential AI deployment with secure enclaves :lock:
https://www.mithrilsecurity.io/
Apache License 2.0
500 stars 35 forks source link

DecodeError: Error parsing message with type 'onnx.ModelProto' #146

Closed loretoparisi closed 1 year ago

loretoparisi commented 1 year ago

I get this error when loading the pretrained whisper model you provide in the Colab example notebook:

import onnx
onnx_model = onnx.load("./whisper_tiny_en_20_tokens.onnx")

Stacktrace:

---------------------------------------------------------------------------
DecodeError                               Traceback (most recent call last)
[<ipython-input-8-d62064a4cf0f>](https://localhost:8080/#) in <module>
      1 import onnx
      2 
----> 3 onnx_model = onnx.load("./whisper_tiny_en_20_tokens.onnx")

2 frames
[/usr/local/lib/python3.8/dist-packages/onnx/__init__.py](https://localhost:8080/#) in _deserialize(s, proto)
    106         )
    107 
--> 108     decoded = cast(Optional[int], proto.ParseFromString(s))
    109     if decoded is not None and decoded != len(s):
    110         raise google.protobuf.message.DecodeError(

DecodeError: Error parsing message with type 'onnx.ModelProto'