diff --git a/README.md b/README.md
index 8f9d30a..7150370 100644
--- a/README.md
+++ b/README.md
@@ -12,7 +12,7 @@ as a proof-of-concept for the 🤗 FlaxVisionEncoderDecoder Framework.
 
 The model can be used as follows:
 
-In PyTorch
+**In PyTorch**
 ```python
 
 import torch
@@ -53,12 +53,7 @@ print(preds)
 
 ```
 
-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
-
+**In Flax**
 ```python
 
 import jax