Unsupported Model: BGEEmbeddings

[maziyarpanahi]

Is there an existing issue for this?

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

Who can help?

@maz

What are you working on?

BGEEmbeddings is not working as a pretrained model in Python

Current Behavior

It fails to work in Python, only works in Scala: Unsupported Model: BGEEmbeddings

Expected Behavior

It should work

Steps To Reproduce

document = DocumentAssembler()\ 
    .setInputCol("text")\ 
    .setOutputCol("document")

embeddings = BGEEmbeddings.pretrained("bge_base", "en")\ 
    .setInputCols("document")\ 
    .setOutputCol("embeddings")

Spark NLP version and Apache Spark

3.5.0
5.2.2

Type of Spark Application

Python Application

Java Version

11

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