RuntimeError: Error(s) in loading state_dict for DeCLUTR: about declutr HOT 2 CLOSED

!pip install -U "git+https://github.com/JohnGiorgi/DeCLUTR.git"

import torch
from declutr import Encoder

# This can be a path on disk to a model you have trained yourself OR
# the name of one of our pretrained models.
pretrained_model_or_path = "declutr-base"

# Prepare some text to embed
text = [
    "A smiling costumed woman is holding an umbrella.",
    "A happy woman in a fairy costume holds an umbrella.",
    "A soccer game with multiple males playing.",
    "Some men are playing a sport.",
]

# Embed the text
cuda_device = torch.cuda.current_device() if torch.cuda.is_available() else -1
encoder = Encoder(pretrained_model_or_path, cuda_device=cuda_device)
embeddings = encoder(text)

from scipy.spatial.distance import cosine

# Compute a semantic similarity via the cosine distance
semantic_sim = 1 - cosine(embeddings[0], embeddings[1])
print(semantic_sim)