Prompt for code summarization using AutoModelForSeq2SeqLM of HuggingFace

salesforce / CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation

https://arxiv.org/abs/2305.07922

BSD 3-Clause "New" or "Revised" License

2.66k stars 391 forks source link

Prompt for code summarization using AutoModelForSeq2SeqLM of HuggingFace #92

Closed Runingtime closed 1 year ago

Runingtime commented 1 year ago

Hi,

May I ask what prefix or suffix to use for code summarization when using AutoModelForSeq2SeqLM of HuggingFace?

yuewang-cuhk commented 1 year ago

Hi there, for code summarization, we suggest to use this CodeT5-base checkpoint that is pretrained in code summarization across 6 PLs from CodeSearchNet in a multi-task setting. No prefix is required for this checkpoint.