"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "66954ea25e342fd451c26ec1c295da0b8692086b"
Unverified Commit c7ad3ff5 authored by Joao Gante's avatar Joao Gante Committed by GitHub
Browse files

Update Marian config default vocabulary size (#19464)

* update marian default vocab size

* also update docstring
parent 9e290804
...@@ -43,7 +43,7 @@ class MarianConfig(PretrainedConfig): ...@@ -43,7 +43,7 @@ class MarianConfig(PretrainedConfig):
Args: Args:
vocab_size (`int`, *optional*, defaults to 50265): vocab_size (`int`, *optional*, defaults to 58101):
Vocabulary size of the Marian model. Defines the number of different tokens that can be represented by the Vocabulary size of the Marian model. Defines the number of different tokens that can be represented by the
`inputs_ids` passed when calling [`MarianModel`] or [`TFMarianModel`]. `inputs_ids` passed when calling [`MarianModel`] or [`TFMarianModel`].
d_model (`int`, *optional*, defaults to 1024): d_model (`int`, *optional*, defaults to 1024):
...@@ -110,7 +110,7 @@ class MarianConfig(PretrainedConfig): ...@@ -110,7 +110,7 @@ class MarianConfig(PretrainedConfig):
def __init__( def __init__(
self, self,
vocab_size=50265, vocab_size=58101,
decoder_vocab_size=None, decoder_vocab_size=None,
max_position_embeddings=1024, max_position_embeddings=1024,
encoder_layers=12, encoder_layers=12,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment