Update README.md
This commit is contained in:
parent
973da61a93
commit
ea737d2858
|
@ -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
|
```python
|
||||||
|
|
||||||
import jax
|
import jax
|
||||||
|
|
Loading…
Reference in New Issue