ImportError: cannot import name 'GPTNeoForCausalLM' from 'transformers' (unknown location)

Hi, I am trying to execute:

from transformers import GPT2Tokenizer, GPTNeoForCausalLM
and I’m getting the above mentioned error.
I’m using torch == 1.8.1 and transformers == 4.4.2. Any help or solution?

GPT-Neo was released in version 4.5 only, so you need to upgrade to that version :wink:

Yeah right, it seems to be that. Thanks