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