diff --git a/mteb/models/jasper_models.py b/mteb/models/jasper_models.py index dcef55d553..b3a3f62516 100644 --- a/mteb/models/jasper_models.py +++ b/mteb/models/jasper_models.py @@ -68,7 +68,7 @@ def encode( config_kwargs={"is_text_encoder": True, "vector_dim": 12288}, model_kwargs={ "attn_implementation": "sdpa", - "torch_dtype": torch.float16, + "torch_dtype": torch.bfloat16, }, trust_remote_code=True, max_seq_length=2048,