From cfca638acbc9361fca43a6fcff33fa512ab5df93 Mon Sep 17 00:00:00 2001 From: Philip May Date: Mon, 31 May 2021 17:54:33 +0200 Subject: [PATCH] Add MT5ForConditionalGeneration as supported arch. to summarization README (#11961) * Add MT5ForConditionalGeneration as supported arch. * Update README.md --- examples/pytorch/summarization/README.md | 1 + examples/pytorch/translation/README.md | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/pytorch/summarization/README.md b/examples/pytorch/summarization/README.md index 8efdfd2248..9c5aff9f47 100644 --- a/examples/pytorch/summarization/README.md +++ b/examples/pytorch/summarization/README.md @@ -29,6 +29,7 @@ For the old `finetune_trainer.py` and related utils, see [`examples/legacy/seq2s - `MarianMTModel` - `PegasusForConditionalGeneration` - `T5ForConditionalGeneration` +- `MT5ForConditionalGeneration` `run_summarization.py` is a lightweight example of how to download and preprocess a dataset from the [🤗 Datasets](https://github.com/huggingface/datasets) library or use your own files (jsonlines or csv), then fine-tune one of the architectures above on it. diff --git a/examples/pytorch/translation/README.md b/examples/pytorch/translation/README.md index 034e83fd13..2af7fb335b 100644 --- a/examples/pytorch/translation/README.md +++ b/examples/pytorch/translation/README.md @@ -29,6 +29,7 @@ For the old `finetune_trainer.py` and related utils, see [`examples/legacy/seq2s - `MarianMTModel` - `PegasusForConditionalGeneration` - `T5ForConditionalGeneration` +- `MT5ForConditionalGeneration` `run_translation.py` is a lightweight examples of how to download and preprocess a dataset from the [🤗 Datasets](https://github.com/huggingface/datasets) library or use your own files (jsonlines or csv), then fine-tune one of the architectures above on it.