facebookresearch / ParlAI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
https://parl.ai
MIT License
10.49k stars 2.1k forks source link

Add DialoGPT as an agent #2741

Closed emilydinan closed 2 years ago

emilydinan commented 4 years ago

Add Microsoft's DialoGPT as an agent in ParlAI: https://github.com/microsoft/DialoGPThttps://github.com/microsoft/DialoGPT

Given that this agent is based off of the Hugging Face transformer's repo, it should go inside this folder: https://github.com/facebookresearch/ParlAI/tree/master/parlai/agents/hugging_face

emilydinan commented 4 years ago

Hugging face seems to have ported it here, so you can probably use the existing GPT2 code with a different constructor: https://huggingface.co/transformers/model_doc/dialogpt.html

github-actions[bot] commented 4 years ago

This issue has not had activity in 30 days. Please feel free to reopen if you have more issues. You may apply the "never-stale" tag to prevent this from happening.