Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
7972f995
Unverified
Commit
7972f995
authored
Oct 14, 2022
by
Sujay
Committed by
GitHub
Oct 14, 2022
Browse files
[Doctest] Add configuration_vision_text_dual_encoder.py (#19580)
* initial commit * few suggested changes
parent
2bd2de62
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
2 deletions
+3
-2
src/transformers/models/vision_text_dual_encoder/configuration_vision_text_dual_encoder.py
...xt_dual_encoder/configuration_vision_text_dual_encoder.py
+2
-2
utils/documentation_tests.txt
utils/documentation_tests.txt
+1
-0
No files found.
src/transformers/models/vision_text_dual_encoder/configuration_vision_text_dual_encoder.py
View file @
7972f995
...
...
@@ -57,7 +57,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig):
>>> 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)
>>> # Accessing the model configuration
...
...
@@ -65,7 +65,7 @@ class VisionTextDualEncoderConfig(PretrainedConfig):
>>> config_text = model.config.text_config
>>> # Saving the model, including its configuration
>>> model.save_pretrained("
my-model
")
>>> model.save_pretrained("
vit-bert
")
>>> # loading model and config from pretrained folder
>>> vision_text_config = VisionTextDualEncoderConfig.from_pretrained("vit-bert")
...
...
utils/documentation_tests.txt
View file @
7972f995
...
...
@@ -118,6 +118,7 @@ src/transformers/models/van/modeling_van.py
src/transformers/models/videomae/modeling_videomae.py
src/transformers/models/vilt/modeling_vilt.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/modeling_vit.py
src/transformers/models/vit/modeling_tf_vit.py
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment