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