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.

Sign up or log in to comment