Hugging Face's logo

google
/
pix2struct-ai2d-base

Visual Question Answering
Transformers
PyTorch
Safetensors
pix2struct
image-text-to-text
Model card Files Community
3
ybelkada commited on
Commit
8b5fd15
•
1 Parent(s): 10fac4e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -50,11 +50,11 @@ This model has been fine-tuned on VQA, you need to provide a question in a speci
50
 
51
  You can use the [`convert_pix2struct_checkpoint_to_pytorch.py`](https://github.com/huggingface/transformers/blob/main/src/transformers/models/pix2struct/convert_pix2struct_checkpoint_to_pytorch.py) script as follows:
52
  ```bash
53
- python convert_pix2struct_checkpoint_to_pytorch.py --t5x_checkpoint_path PATH_TO_T5X_CHECKPOINTS --pytorch_dump_path PATH_TO_SAVE
54
  ```
55
  if you are converting a large model, run:
56
  ```bash
57
- python convert_pix2struct_checkpoint_to_pytorch.py --t5x_checkpoint_path PATH_TO_T5X_CHECKPOINTS --pytorch_dump_path PATH_TO_SAVE --use-large
58
  ```
59
  Once saved, you can push your converted model with the following snippet:
60
  ```python
 
50
 
51
  You can use the [`convert_pix2struct_checkpoint_to_pytorch.py`](https://github.com/huggingface/transformers/blob/main/src/transformers/models/pix2struct/convert_pix2struct_checkpoint_to_pytorch.py) script as follows:
52
  ```bash
53
+ python convert_pix2struct_checkpoint_to_pytorch.py --t5x_checkpoint_path PATH_TO_T5X_CHECKPOINTS --pytorch_dump_path PATH_TO_SAVE --is_vqa
54
  ```
55
  if you are converting a large model, run:
56
  ```bash
57
+ python convert_pix2struct_checkpoint_to_pytorch.py --t5x_checkpoint_path PATH_TO_T5X_CHECKPOINTS --pytorch_dump_path PATH_TO_SAVE --use-large --is_vqa
58
  ```
59
  Once saved, you can push your converted model with the following snippet:
60
  ```python