Skip to content

Commit ace36e3

Browse files
authored
Merge branch 'embeddings-benchmark:main' into adding_the_first_closed_dataset
2 parents 87a831a + 10c4948 commit ace36e3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mteb/models/gritlm_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def gritlm_instruction(instruction: str = "", prompt_type=None) -> str:
3838
memory_usage_mb=13813,
3939
embed_dim=4096,
4040
license="apache-2.0",
41-
max_tokens=4096,
41+
max_tokens=32768,
4242
reference="https://huggingface.co/GritLM/GritLM-7B",
4343
similarity_fn_name="cosine",
4444
framework=["GritLM", "PyTorch"],
@@ -64,7 +64,7 @@ def gritlm_instruction(instruction: str = "", prompt_type=None) -> str:
6464
release_date="2024-02-15",
6565
n_parameters=57_920_000_000,
6666
memory_usage_mb=89079,
67-
embed_dim=4096,
67+
embed_dim=32768,
6868
license="apache-2.0",
6969
max_tokens=4096,
7070
reference="https://huggingface.co/GritLM/GritLM-8x7B",

0 commit comments

Comments
 (0)