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

[Doctest] DeiT Config for doctest (#19560)

Browse Source
This commit is contained in:
Partho
2022-10-13 15:32:40 +05:30
committed by GitHub
parent f4b386765d
commit 4a3578f23f
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/deit/configuration_deit.py
View File

@@ -80,12 +80,12 @@ class DeiTConfig(PretrainedConfig):
Example:
```python
>>> from transformers import DeiTModel, DeiTConfig
>>> from transformers import DeiTConfig, DeiTModel
>>> # Initializing a DeiT deit-base-distilled-patch16-224 style configuration
>>> configuration = DeiTConfig()
>>> # Initializing a model from the deit-base-distilled-patch16-224 style configuration
>>> # Initializing a model (with random weights) from the deit-base-distilled-patch16-224 style configuration
>>> model = DeiTModel(configuration)
>>> # Accessing the model configuration

1
utils/documentation_tests.txt
View File

@@ -36,6 +36,7 @@ src/transformers/models/data2vec/modeling_data2vec_vision.py
src/transformers/models/deberta/modeling_deberta.py
src/transformers/models/deberta_v2/modeling_deberta_v2.py
src/transformers/models/deformable_detr/modeling_deformable_detr.py
src/transformers/models/deit/configuration_deit.py
src/transformers/models/deit/modeling_deit.py
src/transformers/models/deit/modeling_tf_deit.py
src/transformers/models/detr/modeling_detr.py