SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Doctest] Add configuration_vision_text_dual_encoder.py (#19580)

Browse Source 
* initial commit

* few suggested changes
This commit is contained in:
Sujay
2022-10-14 22:15:15 +05:30
committed by GitHub
parent 2bd2de62c9
commit 7972f995b3
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/vision_text_dual_encoder/configuration_vision_text_dual_encoder.py
View File

@@ -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):
>>> 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")

1
utils/documentation_tests.txt
View File

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