From ea737d2858993ecada2a248a5ad1eeb3acb2e7d9 Mon Sep 17 00:00:00 2001 From: Yih-Dar SHIEH Date: Mon, 25 Oct 2021 10:04:59 +0000 Subject: [PATCH] Update README.md --- README.md | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index a09526c..8f9d30a 100644 --- a/README.md +++ b/README.md @@ -53,7 +53,12 @@ print(preds) ``` -In Flax +In order to use Flax's version of this model, currently it is necessary to install a particular branch of `transformers`: +``` +pip install git+https://github.com/ydshieh/transformers.git@flax_vision_encoder_decoder +``` +and it can be used as + ```python import jax