diff --git a/mteb/models/gritlm_models.py b/mteb/models/gritlm_models.py index 9afc5b4c82..e22d8681d4 100644 --- a/mteb/models/gritlm_models.py +++ b/mteb/models/gritlm_models.py @@ -38,7 +38,7 @@ def gritlm_instruction(instruction: str = "", prompt_type=None) -> str: memory_usage_mb=13813, embed_dim=4096, license="apache-2.0", - max_tokens=4096, + max_tokens=32768, reference="https://huggingface.co/GritLM/GritLM-7B", similarity_fn_name="cosine", framework=["GritLM", "PyTorch"], @@ -64,7 +64,7 @@ def gritlm_instruction(instruction: str = "", prompt_type=None) -> str: release_date="2024-02-15", n_parameters=57_920_000_000, memory_usage_mb=89079, - embed_dim=4096, + embed_dim=32768, license="apache-2.0", max_tokens=4096, reference="https://huggingface.co/GritLM/GritLM-8x7B",