How to use Cohere embed v3 in Colab.

#3
by tareeb23 - opened

Can we run embedding model in colab using transformer library directly in colab.
I tried but was getting this error :

code :
from transformers import AutoModel
model = AutoModel.from_pretrained("Cohere/Cohere-embed-multilingual-v3.0")

Error :
/usr/local/lib/python3.10/dist-packages/transformers/models/auto/configuration_auto.py in from_pretrained(cls, pretrained_model_name_or_path, **kwargs)
958 return CONFIG_MAPPING[pattern].from_dict(config_dict, **unused_kwargs)
959
--> 960 raise ValueError(
961 f"Unrecognized model in {pretrained_model_name_or_path}. "
962 f"Should have a model_type key in its {CONFIG_NAME}, or contain one of the following strings "

ValueError: Unrecognized model in Cohere/Cohere-embed-multilingual-v3.0. Should have a model_type key in its config.json, or contain one of the following strings in its name: albert, align, altclip, audio-spectrogram-transformer, autoformer, bark, bart, beit, bert ....................

tareeb23 changed discussion title from How to use Cohere v3 in Colab. to How to use Cohere embed v3 in Colab.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment