Unverified Commit 7972f995 authored by Sujay's avatar Sujay Committed by GitHub
Browse files

[Doctest] Add configuration_vision_text_dual_encoder.py (#19580)

* initial commit

* few suggested changes
parent 2bd2de62
...@@ -57,7 +57,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig): ...@@ -57,7 +57,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig):
>>> config = VisionTextDualEncoderConfig.from_vision_text_configs(config_vision, config_text, projection_dim=512) >>> config = VisionTextDualEncoderConfig.from_vision_text_configs(config_vision, config_text, projection_dim=512)
>>> # Initializing a BERT and ViT model >>> # Initializing a BERT and ViT model (with random weights)
>>> model = VisionTextDualEncoderModel(config=config) >>> model = VisionTextDualEncoderModel(config=config)
>>> # Accessing the model configuration >>> # Accessing the model configuration
...@@ -65,7 +65,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig): ...@@ -65,7 +65,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig):
>>> config_text = model.config.text_config >>> config_text = model.config.text_config
>>> # Saving the model, including its configuration >>> # Saving the model, including its configuration
>>> model.save_pretrained("my-model") >>> model.save_pretrained("vit-bert")
>>> # loading model and config from pretrained folder >>> # loading model and config from pretrained folder
>>> vision_text_config = VisionTextDualEncoderConfig.from_pretrained("vit-bert") >>> vision_text_config = VisionTextDualEncoderConfig.from_pretrained("vit-bert")
......
...@@ -118,6 +118,7 @@ src/transformers/models/van/modeling_van.py ...@@ -118,6 +118,7 @@ src/transformers/models/van/modeling_van.py
src/transformers/models/videomae/modeling_videomae.py src/transformers/models/videomae/modeling_videomae.py
src/transformers/models/vilt/modeling_vilt.py src/transformers/models/vilt/modeling_vilt.py
src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py src/transformers/models/vision_encoder_decoder/modeling_vision_encoder_decoder.py
src/transformers/models/vision_text_dual_encoder/configuration_vision_text_dual_encoder.py
src/transformers/models/vit/configuration_vit.py src/transformers/models/vit/configuration_vit.py
src/transformers/models/vit/modeling_vit.py src/transformers/models/vit/modeling_vit.py
src/transformers/models/vit/modeling_tf_vit.py src/transformers/models/vit/modeling_tf_vit.py
......
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