From 0a6193252ee98fe4c4326cb971191c7c03fd93a8 Mon Sep 17 00:00:00 2001 From: sha016 <92833633+sha016@users.noreply.github.com> Date: Mon, 28 Nov 2022 08:47:45 -0600 Subject: [PATCH] [Doctest] Add configuration_fsmt.py (#19936) * fsmt doctest * Update Co-authored-by: ydshieh --- src/transformers/models/fsmt/configuration_fsmt.py | 8 +++++++- utils/documentation_tests.txt | 1 + 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/src/transformers/models/fsmt/configuration_fsmt.py b/src/transformers/models/fsmt/configuration_fsmt.py index de96c768a2..85775d07b2 100644 --- a/src/transformers/models/fsmt/configuration_fsmt.py +++ b/src/transformers/models/fsmt/configuration_fsmt.py @@ -125,8 +125,14 @@ class FSMTConfig(PretrainedConfig): ```python >>> from transformers import FSMTConfig, FSMTModel - >>> config = FSMTConfig.from_pretrained("facebook/wmt19-en-ru") + >>> # Initializing a FSMT facebook/wmt19-en-ru style configuration + >>> config = FSMTConfig() + + >>> # Initializing a model (with random weights) from the configuration >>> model = FSMTModel(config) + + >>> # Accessing the model configuration + >>> configuration = model.config ```""" model_type = "fsmt" attribute_map = {"num_attention_heads": "encoder_attention_heads", "hidden_size": "d_model"} diff --git a/utils/documentation_tests.txt b/utils/documentation_tests.txt index 07af79e535..9293dc3c39 100644 --- a/utils/documentation_tests.txt +++ b/utils/documentation_tests.txt @@ -79,6 +79,7 @@ src/transformers/models/electra/modeling_tf_electra.py src/transformers/models/ernie/configuration_ernie.py src/transformers/models/flava/configuration_flava.py src/transformers/models/fnet/configuration_fnet.py +src/transformers/models/fsmt/configuration_fsmt.py src/transformers/models/glpn/modeling_glpn.py src/transformers/models/gpt2/configuration_gpt2.py src/transformers/models/gpt2/modeling_gpt2.py