Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1,110 changes: 1,110 additions & 0 deletions mteb/abstasks/AbsTaskRTEB.py

Large diffs are not rendered by default.

1 change: 1 addition & 0 deletions mteb/abstasks/TaskMetadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -126,6 +126,7 @@
"Summarization",
"InstructionRetrieval",
"Speed",
"RTEB",
) + MIEB_TASK_TYPE

TASK_TYPE = Literal[TASK_TYPE]
Expand Down
1 change: 1 addition & 0 deletions mteb/abstasks/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
from .AbsTaskPairClassification import *
from .AbsTaskReranking import *
from .AbsTaskRetrieval import *
from .AbsTaskRTEB import *
from .AbsTaskSpeedTask import *
from .AbsTaskSTS import *
from .AbsTaskSummarization import *
Expand Down
25 changes: 25 additions & 0 deletions mteb/models/voyage_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -368,6 +368,31 @@ def _batched_encode(
public_training_data=None,
)

voyage_3_large = ModelMeta(
name="voyageai/voyage-3-large", # Use the identifier the user provided
revision="1", # Assuming revision 1
release_date="2024-09-18", # Assuming same release as voyage-3
languages=None,
loader=partial( # type: ignore
VoyageWrapper,
model_name="voyage-3-large", # Match the API model name
model_prompts=model_prompts,
),
max_tokens=32000, # Assuming same as voyage-3
embed_dim=1024, # Assuming same as voyage-3
open_weights=False,
n_parameters=None,
memory_usage_mb=None,
license=None,
reference="https://blog.voyageai.com/2024/09/18/voyage-3/", # Assuming same reference
similarity_fn_name="cosine",
framework=["API"],
use_instructions=True,
training_datasets=VOYAGE_TRAINING_DATA,
public_training_code=None,
public_training_data=None,
)

voyage_3_lite = ModelMeta(
name="voyageai/voyage-3-lite",
revision="1",
Expand Down
Empty file removed mteb/rteb/__init__.py
Empty file.
Empty file removed mteb/rteb/ebr/__init__.py
Empty file.
220 changes: 0 additions & 220 deletions mteb/rteb/ebr/__main__.py

This file was deleted.

1 change: 0 additions & 1 deletion mteb/rteb/ebr/core/__init__.py

This file was deleted.

1 change: 0 additions & 1 deletion mteb/rteb/ebr/core/base/__init__.py

This file was deleted.

76 changes: 0 additions & 76 deletions mteb/rteb/ebr/core/base/dataset.py

This file was deleted.

Loading