Fail to initialize model using AutoModel.from_config()
#10
by AppleSwing - opened
Hi, I am trying to initialize the model from config which is generated by AutoConfig.from_pretrained(). However it returns: "name": "ValueError",
"message": "Unrecognized configuration class <class 'transformers_modules.NV-Embed-v2.configuration_nvembed.LatentAttentionConfig'> for this kind of AutoModel: AutoModel
Hi, @AppleSwing . Please refer the appropriate package installation: https://huggingface.co/nvidia/NV-Embed-v2#2-required-packages
nada5 changed discussion status to closed