ydshieh
/
vit-gpt2-coco-en
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

This commit is contained in:
Yih-Dar SHIEH 2021-11-10 18:59:24 +00:00 committed by huggingface-web
parent ea737d2858
commit b0acf67ee3
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@ -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