diff --git a/mteb/models/e5_instruct.py b/mteb/models/e5_instruct.py index e27fc99b77..5bc436bf15 100644 --- a/mteb/models/e5_instruct.py +++ b/mteb/models/e5_instruct.py @@ -7,7 +7,7 @@ from mteb.model_meta import ModelMeta from mteb.models.e5_models import ( E5_PAPER_RELEASE_DATE, - E5_TRAINING_DATA, + ME5_TRAINING_DATA, XLMR_LANGUAGES, ) from mteb.models.instruct_wrapper import instruct_wrapper @@ -17,7 +17,7 @@ E5_INSTRUCTION = "Instruct: {instruction}\nQuery: " E5_MISTRAL_TRAINING_DATA = { - **E5_TRAINING_DATA, + **ME5_TRAINING_DATA, "FEVER": ["train"], "FEVERHardNegatives": ["train"], "FEVER-NL": ["train"], # translation not trained on @@ -57,7 +57,7 @@ max_tokens=514, public_training_code=None, public_training_data=None, - training_datasets=E5_TRAINING_DATA, + training_datasets=ME5_TRAINING_DATA, ) e5_mistral = ModelMeta(