center-for-curriculum-redesign/bge_1-5_query_embeddings

center-for-curriculum-redesign / bge_1-5_query_embeddings

Query embedding generator for BAAI's bge-large-en v1.5 embedding model

Public
5.4K runs
GitHub
Paper
License

Run time and cost

This model costs approximately $0.015 to run on Replicate, or 66 runs per $1, but this varies depending on your inputs. It is also open source and you can run it on your own computer with Docker.

This model runs on Nvidia A40 GPU hardware. Predictions typically complete within 26 seconds. The predict time for this model varies significantly based on the inputs.

Readme

BAAI’s (english) embedding model version 1.5 for passage retrieval and sentence similarity, using the Center For Curriculum Redesign’s Replicate Embedding Cog template.

As of September 2023, this model is state of the art.

output dimensions : 1024
parameters: 326M
MTEB score average: 64.23

Dimension	Sequence Length	Average (56)	Retrieval (15)	Clustering (11)	Pair Classification (3)	Reranking (4)	STS (10)	Summarization (1)	Classification (12)
1024	512	64.23	54.29	46.08	87.12	60.03	83.11	31.61	75.97

Usage

See the examples tab.