...     "researchers was the fact that the unicorns spoke perfect English."
... )
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
gen_tokens = model.generate(
...     input_ids,
...     do_sample=True,
...     temperature=0.9,
...     max_length=100,
... )
gen_text = tokenizer.batch_decode(gen_tokens)[0]
```
GPTNeoConfig
[[autodoc]] GPTNeoConfig
GPTNeoModel
[[autodoc]] GPTNeoModel
    - forward
GPTNeoForCausalLM
[[autodoc]] GPTNeoForCausalLM
    - forward
GPTNeoForSequenceClassification
[[autodoc]] GPTNeoForSequenceClassification
    - forward
FlaxGPTNeoModel
[[autodoc]] FlaxGPTNeoModel
    - call
FlaxGPTNeoForCausalLM
[[autodoc]] FlaxGPTNeoForCausalLM
    - call
