Onnx models fail when saving transformer

[mehmetbutgul]

Is there an existing issue for this?

• I have searched the existing issues and did not find a match.

Who can help?

@maziyarpanahi

What are you working on?

I am working on the Colab and local.
I want to save a transformer that pretrained the Onnx model. But, I have faced an error.
When I investigate the error, I have noticed that the main problem is serializing a onnx model -->

spark-nlp/src/main/scala/com/johnsnowlabs/ml/onnx/OnnxSerializeModel.scala

Line 31 in 43aa4b0

def writeOnnxModels(

Current Behavior

Error while saving the transformer,
This error applies to most ONNX based annotators

Expected Behavior

writing the transformer stage to file

Steps To Reproduce

DeBertaEmbeddings.pretrained("deberta_embeddings_erlangshen_v2_chinese_sentencepiece","zh") \
     .write().overwrite().save("models/deberta_model")

Colab notebook: https://colab.research.google.com/drive/119u6hXoT1PRB9F38InuEV-bm4g1uu9UH?usp=sharing

Spark NLP version and Apache Spark

spark-nlp==5.3.1
pyspark

Type of Spark Application

No response

Java Version

No response

Java Home Directory

No response

Setup and installation

No response

Operating System and Version

No response

Link to your project (if available)

No response

Additional Information

No response