Wav2vec2 and hubert export to onnx

Hi, I am trying to export wav2vec and hubert models to onnx runtime. Exporting seems to work fine using your transformers.onnx or by doing it “manually”. But only if the attention mask is None. Do you have any pointers as to how the model code could be altered so that loading doesn’t fail with errors like:

 Type Error: Type 'tensor(bool)' of input parameter (744) of operator (ScatterND) in node (ScatterND_309) is invalid

if a mask is passed?