ydshieh
/

vit-gpt2-coco-en

vision-encoder-decoder

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

ydshieh HF staff commited on Nov 10, 2021

Commit

b0acf67

•

1 Parent(s): ea737d2

Update README.md

Files changed (1) hide show

README.md +2 -7

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ as a proof-of-concept for the 🤗 FlaxVisionEncoderDecoder Framework.
 The model can be used as follows:
-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
 ```python
 import jax

 The model can be used as follows:
+**In PyTorch**
 ```python
 import torch
 ```
+**In Flax**
 ```python
 import jax