Skip to content

Latest commit

 

History

History
40 lines (29 loc) · 1.47 KB

README.md

File metadata and controls

40 lines (29 loc) · 1.47 KB
title emoji colorFrom colorTo sdk sdk_version app_file pinned tags
MTEB Arena
⚔️
indigo
blue
gradio
4.21.0
app.py
false
arena
leaderboard

MTEB Arena

Setup

  1. Install PyTorch according to your system
  2. pip install -r requirements.txt

Run

python app.py

If you'd like to debug the app, set DEBUG=True in the beginning of app.py, which will use fewer models and can be easily run locally on CPUs. Also you may want to set GCP_INDEX=False and just use the local indices for the two debug models.

Some models require API keys which you can set as environment variables, e.g. VOYAGE_API_KEY=YOUR_KEY HF_TOKEN=YOUR_KEY OPENAI_API_KEY=YOUR_KEY CO_API_KEY=YOUR_KEY python app.py

Adding a Model

  1. Add the model to mteb via PR.
  2. Add the model to this repo's models.py & models_meta.yml via PR.
  3. Test it in the Clustering and STS interfaces at least. Ideally, also retrieval.
  4. We will then consider adding it to the arena. Please note that keeping a model in the arena is costly to us as (1) we need to keep constant GPUs running with it in memory (2) we need to maintain retrieval indices for it. So we will focus on models that are most useful to the community. However, if you are willing to sponsor the hosting of your model, we may be able to add it faster.

Results

Results are auto-saved to mteb/arena-results.