Transformers: model.generate source code

[chmod-700]

Hi there,

I am trying to use BART to do NLG task. During my reading the BART tutorial on the website, I couldn't find the definition of 'model.generate()" function. Could you please add some explaination on that? thanks in advance

[mariamabarham]

generate() is defined in the PretrainedModel class here https://github.com/huggingface/transformers/blob/master/src/transformers/modeling_utils.py#L872. Given that BartModel, BartForConditionalGeneration, etc, inherit from PretrainedModel they all have access to this generate function.

Hope this help?

[yjernite]

We actually just moved all of the generate functions to their own file:
https://github.com/huggingface/transformers/blob/c4d4e8bdbd25d9463d41de6398940329c89b7fb6/src/transformers/generation_utils.py#L101

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.