diff --git a/mteb/models/openai_models.py b/mteb/models/openai_models.py index 50598f9eee..6d3ce18aa2 100644 --- a/mteb/models/openai_models.py +++ b/mteb/models/openai_models.py @@ -27,6 +27,7 @@ def __init__( max_tokens: int, tokenizer_name: str = "cl100k_base", embed_dim: int | None = None, + client: Any | None = None, # OpenAI **kwargs, ) -> None: """Wrapper for OpenAIs embedding API. @@ -49,7 +50,7 @@ def __init__( ) import tiktoken - self._client = OpenAI() + self._client = OpenAI() if client is None else client self._model_name = model_name if embed_dim is None: diff --git a/mteb/models/overview.py b/mteb/models/overview.py index d5ea6a5c55..be4acc6b08 100644 --- a/mteb/models/overview.py +++ b/mteb/models/overview.py @@ -38,6 +38,7 @@ e5_v, evaclip_models, fa_models, + geogpt_models, gme_v_models, google_models, gritlm_models, @@ -91,7 +92,6 @@ voyage_models, voyage_v, xyz_models, - geogpt_models, ) logger = logging.getLogger(__name__)