From 02039352b2d78a98ab7cba1cf805f0d394e84a8f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 1 Sep 2021 09:50:21 +0200 Subject: [PATCH] Update README.md --- templates/adding_a_new_model/README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/templates/adding_a_new_model/README.md b/templates/adding_a_new_model/README.md index 8b2d03e056..d19270bf75 100644 --- a/templates/adding_a_new_model/README.md +++ b/templates/adding_a_new_model/README.md @@ -17,7 +17,7 @@ limitations under the License. # Using `cookiecutter` to generate models This folder contains templates to generate new models that fit the current API and pass all tests. It generates -models in both PyTorch and TensorFlow, completes the `__init__.py` and auto-modeling files, and creates the +models in both PyTorch, TensorFlow, and Flax and completes the `__init__.py` and auto-modeling files, and creates the documentation. ## Usage @@ -117,4 +117,4 @@ will be merged quickly: - You should run the `make fixup` utility to fix the style of the files and to ensure the code quality meets the library's standards. - You should complete the documentation file (`docs/source/model_doc/.rst`) so that your model may be - usable. \ No newline at end of file + usable.