Update README.md
Browse files
README.md
CHANGED
|
@@ -53,7 +53,12 @@ print(preds)
|
|
| 53 |
|
| 54 |
```
|
| 55 |
|
| 56 |
-
In Flax
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 57 |
```python
|
| 58 |
|
| 59 |
import jax
|
|
|
|
| 53 |
|
| 54 |
```
|
| 55 |
|
| 56 |
+
In order to use Flax's version of this model, currently it is necessary to install a particular branch of `transformers`:
|
| 57 |
+
```
|
| 58 |
+
pip install git+https://github.com/ydshieh/transformers.git@flax_vision_encoder_decoder
|
| 59 |
+
```
|
| 60 |
+
and it can be used as
|
| 61 |
+
|
| 62 |
```python
|
| 63 |
|
| 64 |
import jax
|