Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ as a proof-of-concept for the 🤗 FlaxVisionEncoderDecoder Framework.
|
|
12 |
|
13 |
The model can be used as follows:
|
14 |
|
15 |
-
In PyTorch
|
16 |
```python
|
17 |
|
18 |
import torch
|
@@ -53,12 +53,7 @@ print(preds)
|
|
53 |
|
54 |
```
|
55 |
|
56 |
-
In
|
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
|
|
|
12 |
|
13 |
The model can be used as follows:
|
14 |
|
15 |
+
**In PyTorch**
|
16 |
```python
|
17 |
|
18 |
import torch
|
|
|
53 |
|
54 |
```
|
55 |
|
56 |
+
**In Flax**
|
|
|
|
|
|
|
|
|
|
|
57 |
```python
|
58 |
|
59 |
import jax
|