From 25b0463d0ba3fcbcf7fff8aa4027a2d8e959364b Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Tue, 22 Sep 2020 13:09:35 -0400 Subject: [PATCH] [s2s] add supported architecures to MD (#7252) --- examples/seq2seq/README.md | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/examples/seq2seq/README.md b/examples/seq2seq/README.md index 2eb9c6e412..03e8b0f9ef 100644 --- a/examples/seq2seq/README.md +++ b/examples/seq2seq/README.md @@ -1,10 +1,18 @@ ## Sequence to Sequence This directory contains examples for finetuning and evaluating transformers on summarization and translation tasks. -Summarization support is more mature than translation support. Please tag @sshleifer with any issues/unexpected behaviors, or send a PR! For `bertabs` instructions, see [`bertabs/README.md`](bertabs/README.md). +### Supported Architectures + +- `BartForConditionalGeneration` (and anything that inherits from it) +- `MarianMTModel` +- `PegasusForConditionalGeneration` +- `MBartForConditionalGeneration` +- `FSMTForConditionalGeneration` +- `T5ForConditionalGeneration` + ## Datasets