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

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

Browse Source
This commit is contained in:
Sujay
2022-10-13 22:37:10 +05:30
committed by GitHub
parent 1f6a28c71c
commit 3d320c78c3
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/trajectory_transformer/configuration_trajectory_transformer.py
View File

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

1
utils/documentation_tests.txt
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/speech_to_text_2/modeling_speech_to_text_2.py
src/transformers/models/segformer/modeling_tf_segformer.py src/transformers/models/segformer/modeling_tf_segformer.py
src/transformers/models/swin/modeling_swin.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/trocr/modeling_trocr.py
src/transformers/models/unispeech/modeling_unispeech.py src/transformers/models/unispeech/modeling_unispeech.py
src/transformers/models/unispeech_sat/modeling_unispeech_sat.py src/transformers/models/unispeech_sat/modeling_unispeech_sat.py