From 9bb26f2505921bb6888f7d7d0b5676e3becb38b7 Mon Sep 17 00:00:00 2001 From: Thomas Date: Mon, 17 Oct 2022 16:53:36 +0800 Subject: [PATCH] [Doctest] Add `configuration_trocr.py` (#19658) * trocr Config for doctest * ran make style --- src/transformers/models/trocr/configuration_trocr.py | 4 ++-- utils/documentation_tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/trocr/configuration_trocr.py b/src/transformers/models/trocr/configuration_trocr.py index d27b032ebe..ad22fbbe0f 100644 --- a/src/transformers/models/trocr/configuration_trocr.py +++ b/src/transformers/models/trocr/configuration_trocr.py @@ -82,12 +82,12 @@ class TrOCRConfig(PretrainedConfig): Example: ```python - >>> from transformers import TrOCRForCausalLM, TrOCRConfig + >>> from transformers import TrOCRConfig, TrOCRForCausalLM >>> # Initializing a TrOCR-base style configuration >>> configuration = TrOCRConfig() - >>> # Initializing a model from the TrOCR-base style configuration + >>> # Initializing a model (with random weights) from the TrOCR-base style configuration >>> model = TrOCRForCausalLM(configuration) >>> # Accessing the model configuration diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 1069c3b6a5..6cbc0958d8 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -114,6 +114,7 @@ src/transformers/models/swinv2/configuration_swinv2.py src/transformers/models/time_series_transformer/configuration_time_series_transformer.py src/transformers/models/time_series_transformer/modeling_time_series_transformer.py src/transformers/models/trajectory_transformer/configuration_trajectory_transformer.py +src/transformers/models/trocr/configuration_trocr.py src/transformers/models/trocr/modeling_trocr.py src/transformers/models/unispeech/configuration_unispeech.py src/transformers/models/unispeech/modeling_unispeech.py