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): ...@@ -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. >>> # initialize a model from pretrained ViT and BERT models. Note that the projection layers will be randomly initialized.
>>> model = FlaxVisionTextDualEncoderModel.from_vision_text_pretrained( >>> 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 >>> # saving model after fine-tuning
>>> model.save_pretrained("./vit-bert") >>> model.save_pretrained("./vit-bert")
......
...@@ -469,7 +469,7 @@ class VisionTextDualEncoderModel(PreTrainedModel): ...@@ -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. >>> # initialize a model from pretrained ViT and BERT models. Note that the projection layers will be randomly initialized.
>>> model = VisionTextDualEncoderModel.from_vision_text_pretrained( >>> 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 >>> # saving model after fine-tuning
>>> model.save_pretrained("./vit-bert") >>> 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