diff --git a/src/transformers/models/esm/configuration_esm.py b/src/transformers/models/esm/configuration_esm.py index 9634a20015f2..083664747ddd 100644 --- a/src/transformers/models/esm/configuration_esm.py +++ b/src/transformers/models/esm/configuration_esm.py @@ -87,11 +87,14 @@ class EsmConfig(PretrainedConfig): ```python >>> from transformers import EsmModel, EsmConfig - >>> # Initializing a ESM facebook/esm-1b style configuration >>> configuration = EsmConfig() + >>> # Initializing a ESM facebook/esm-1b style configuration + >>> configuration = EsmConfig(vocab_size=33) - >>> # Initializing a model from the configuration >>> model = ESMModel(configuration) + >>> # Initializing a model from the configuration + >>> model = EsmModel(configuration) - >>> # Accessing the model configuration >>> configuration = model.config + >>> # Accessing the model configuration + >>> configuration = model.config ```""" model_type = "esm"