Viewing a single comment thread. View all comments

GasZealousideal8691 OP t1_j4gpu8j wrote

GPT Neo is GPTNeoForCausalLM, and GPT2 is GPT2LMHeadModel. Like I said, I am not 100% familiar with these, but the huggingface docs listed both as “GPT-neo/GPT2 with an LM head”, so I figured they were analogous.

1