How to use BAAI bge 1.5 Models in Marqo?

Support
1

I want to use the BAAI/bge-base-en-v1.5 model in Marqo because it looks like it ranks high on MTEB for retrieval. How do I do this?

1 Like
2

We don’t maintain a list of every possible model, but any text model from huggingface that can be loaded with SentenceTransformer from sentence_transformers should work. As such the list of possible models extends to any model within that API. There are some more details on loading custom models under section 3.2 of the model reference.

For the BAAI model you are after you could use the following index defaults configuration:

{
    "index_defaults": {
        "model_properties": {
            "name": "BAAI/bge-base-en-v1.5",
            "type": "hf",
            "dimensions": 768,
        },
        "model": "baai-bge-base-en-v1.5",
        "normalize_embeddings": True,
    }
}
1 Like