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

[Doctest] Add configuration_vit.py (#19561)

Browse Source 
* ViT Config for doctest
This commit is contained in:
Partho
2022-10-13 15:37:14 +05:30
committed by GitHub
parent f173ceefc0
commit 0ae3ec5b9d
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/vit/configuration_vit.py
View File

@@ -77,12 +77,12 @@ class ViTConfig(PretrainedConfig):
Example:
```python
>>> from transformers import ViTModel, ViTConfig
>>> from transformers import ViTConfig, ViTModel
>>> # Initializing a ViT vit-base-patch16-224 style configuration
>>> configuration = ViTConfig()
>>> # Initializing a model from the vit-base-patch16-224 style configuration
>>> # Initializing a model (with random weights) from the vit-base-patch16-224 style configuration
>>> model = ViTModel(configuration)
>>> # Accessing the model configuration

1
utils/documentation_tests.txt
View File

@@ -99,6 +99,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/vit/configuration_vit.py
src/transformers/models/vit/modeling_vit.py
src/transformers/models/vit/modeling_tf_vit.py
src/transformers/models/vit_mae/modeling_vit_mae.py