dunzhang

/

stella_en_400M_v5

like 82

while running the same code below for bge-large-en-v1.55 and gte-large-en-v1.5. I didn't faced any error. But for this model I am facing the error

from sentence_transformers import SentenceTransformer
from sentence_transformers.util import cos_sim
from tqdm.autonotebook import tqdm, trange
import torch 
device = torch.device("cpu")
sentences = ['That is a happy person', 'That is a very happy person']

model = SentenceTransformer('dunzhang/stella_en_400M_v5', 
                            trust_remote_code=True,
                           device ="cpu"
                            )
embeddings = model.encode(sentences)
print(cos_sim(embeddings[0], embeddings[1]))