[Doctest] adds trajectory_transformer config to Docs test (#19586)

This commit is contained in:
Sujay 2022-10-13 22:37:10 +05:30 committed by GitHub
parent 1f6a28c71c
commit 3d320c78c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -93,12 +93,12 @@ class TrajectoryTransformerConfig(PretrainedConfig):
Example:
```python
>>> from transformers import TrajectoryTransformerModel, TrajectoryTransformerConfig
>>> from transformers import TrajectoryTransformerConfig, TrajectoryTransformerModel
>>> # Initializing a TrajectoryTransformer CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
>>> configuration = TrajectoryTransformerConfig()
>>> # Initializing a model from the CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
>>> # Initializing a model (with random weights) from the CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
>>> model = TrajectoryTransformerModel(configuration)
>>> # Accessing the model configuration

View File

@ -96,6 +96,7 @@ src/transformers/models/speech_to_text/modeling_speech_to_text.py
src/transformers/models/speech_to_text_2/modeling_speech_to_text_2.py
src/transformers/models/segformer/modeling_tf_segformer.py
src/transformers/models/swin/modeling_swin.py
src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py
src/transformers/models/trocr/modeling_trocr.py
src/transformers/models/unispeech/modeling_unispeech.py
src/transformers/models/unispeech_sat/modeling_unispeech_sat.py