Possible to include safetensors file at root level?

#2
by shreyask - opened

since this is a static embedding model I was curious to try it with model2vec

from model2vec import StaticModel

# Load a model from the HuggingFace hub (in this case the potion-base-8M model)
model = StaticModel.from_pretrained(
    "sentence-transformers/static-similarity-mrl-multilingual-v1"
)

# Make embeddings
embeddings = model.encode(
    ["It's dangerous to go alone!", "It's a secret to everybody."]
)

# Make sequences of token embeddings
token_embeddings = model.encode_as_sequence(
    ["It's dangerous to go alone!", "It's a secret to everybody."]
)

but getting this 404 error

EntryNotFoundError: 404 Client Error. (Request ID: Root=1-67aed12a-77ea6b143c69fe1b38205ea2;d46ca859-c4c1-4641-ba41-59d410bf7676)

Entry Not Found for url: https://huggingface.co/sentence-transformers/static-similarity-mrl-multilingual-v1/resolve/main/model.safetensors.

If this is something related to model2vec specifically happy to open a PR there in the repo too.

nvm found out what I was missing

model = StaticModel.from_sentence_transformers(
    "sentence-transformers/static-similarity-mrl-multilingual-v1"
) 

this made it work.

shreyask changed discussion status to closed

Sign up or log in to comment