"tests/models/layoutlmv2/test_tokenization_layoutlmv2.py" did not exist on "d33dc7966a8c7f04bbca7ae0ced75cbf26c38d9e"
Unverified Commit 2ca62683 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix from_vision_text_pretrained doc example (#15453)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent dc05dd53
......@@ -449,7 +449,7 @@ class FlaxVisionTextDualEncoderModel(FlaxPreTrainedModel):
>>> # initialize a model from pretrained ViT and BERT models. Note that the projection layers will be randomly initialized.
>>> model = FlaxVisionTextDualEncoderModel.from_vision_text_pretrained(
... "bert-base-uncased", "google/vit-base-patch16-224"
... "google/vit-base-patch16-224", "bert-base-uncased"
... )
>>> # saving model after fine-tuning
>>> model.save_pretrained("./vit-bert")
......
......@@ -469,7 +469,7 @@ class VisionTextDualEncoderModel(PreTrainedModel):
>>> # initialize a model from pretrained ViT and BERT models. Note that the projection layers will be randomly initialized.
>>> model = VisionTextDualEncoderModel.from_vision_text_pretrained(
... "bert-base-uncased", "google/vit-base-patch16-224"
... "google/vit-base-patch16-224", "bert-base-uncased"
... )
>>> # saving model after fine-tuning
>>> model.save_pretrained("./vit-bert")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment