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

[Doctests] Config files for ViTMAE and YOSO (#19567)

Browse Source
This commit is contained in:
KaranGarg
2022-10-13 18:35:02 +05:30
committed by GitHub
parent 05a287ec1a
commit 7180e17256
3 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/models/vit_mae/configuration_vit_mae.py
View File

@@ -82,12 +82,12 @@ class ViTMAEConfig(PretrainedConfig):
Example: Example:
```python ```python
>>> from transformers import ViTMAEModel, ViTMAEConfig >>> from transformers import ViTMAEConfig, ViTMAEModel
>>> # Initializing a ViT MAE vit-mae-base style configuration >>> # Initializing a ViT MAE vit-mae-base style configuration
>>> configuration = ViTMAEConfig() >>> configuration = ViTMAEConfig()
>>> # Initializing a model from the vit-mae-base style configuration >>> # Initializing a model (with random weights) from the vit-mae-base style configuration
>>> model = ViTMAEModel(configuration) >>> model = ViTMAEModel(configuration)
>>> # Accessing the model configuration >>> # Accessing the model configuration

4
src/transformers/models/yoso/configuration_yoso.py
View File

@@ -83,12 +83,12 @@ class YosoConfig(PretrainedConfig):
Example: Example:
```python ```python
>>> from transformers import YosoModel, YosoConfig >>> from transformers import YosoConfig, YosoModel
>>> # Initializing a YOSO uw-madison/yoso-4096 style configuration >>> # Initializing a YOSO uw-madison/yoso-4096 style configuration
>>> configuration = YosoConfig() >>> configuration = YosoConfig()
>>> # Initializing a model from the uw-madison/yoso-4096 style configuration >>> # Initializing a model (with random weights) from the uw-madison/yoso-4096 style configuration
>>> model = YosoModel(configuration) >>> model = YosoModel(configuration)
>>> # Accessing the model configuration >>> # Accessing the model configuration

2
utils/documentation_tests.txt
View File

@@ -106,6 +106,7 @@ 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
src/transformers/models/vit_mae/modeling_vit_mae.py src/transformers/models/vit_mae/modeling_vit_mae.py
src/transformers/models/vit_mae/configuration_vit_mae.py
src/transformers/models/vit_msn/modeling_vit_msn.py src/transformers/models/vit_msn/modeling_vit_msn.py
src/transformers/models/wav2vec2/modeling_wav2vec2.py src/transformers/models/wav2vec2/modeling_wav2vec2.py
src/transformers/models/wav2vec2/tokenization_wav2vec2.py src/transformers/models/wav2vec2/tokenization_wav2vec2.py
@@ -118,3 +119,4 @@ src/transformers/models/whisper/modeling_tf_whisper.py
src/transformers/models/yolos/configuration_yolos.py src/transformers/models/yolos/configuration_yolos.py
src/transformers/models/yolos/modeling_yolos.py src/transformers/models/yolos/modeling_yolos.py
src/transformers/models/x_clip/modeling_x_clip.py src/transformers/models/x_clip/modeling_x_clip.py
src/transformers/models/yoso/configuration_yoso.py