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
DimensionSequence LengthAverage (56)Retrieval (15)Clustering (11)Pair Classification (3)Reranking (4)STS (10)Summarization (1)Classification (12)
102451264.2354.2946.0887.1260.0383.1131.6175.97

Usage


See the examples tab.