Loading via AutoModel gives an error

#9
by shantanuagarwal - opened

When trying to load microsoft/Phi-3-medium-4k-instruct via AutoModel.from_pretrained, I am getting the following error:

ValueError: Unrecognized configuration class <class 'transformers_modules.microsoft.Phi-3-medium-4k-instruct.d194e4e74ffad5a5e193e26af25bcfc80c7f1ffc.configuration_phi3.Phi3Config'> for this kind of AutoModel: AutoModel.
...

I do not get this error when loading via AutoModelForCausalLM.from_pretrained.

Was wondering why we are not able to load the model using AutoModel.
The version of transformers I am using is:

$ conda list | grep transformers
transformers              4.40.2                   pypi_0    pypi

Do I need to update my env?

Sign up or log in to comment