GPTJModel documentation

Hello, I’ve looked at the current documentation on GPTJModel and affiliates
Nonetheless, I can’t find any information on the different functions that can be apply to those models, for instance, I’m trying to find a documentation about the function generate() and I would love to know if other functions can be apply to those model.

Thank you in advance for your answers !

Hi @polynomds1, you can find the docs for the generate function here, and we also have a guide on text generation strategies if you’re interested!

The GPTJ/Neo docs also have a generation section, but perhaps we can also create a generate section for the other GPT models.

1 Like

Thanks @stevhliu, I was also wondering if there was other function like generate that could be apply to those models ?