"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ca1f1c8685a1f8a16bd578a83239d46e1c9a9f7b"
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):
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
`inputs_ids` passed when calling [`MarianModel`] or [`TFMarianModel`].
d_model (`int`, *optional*, defaults to 1024):
......@@ -110,7 +110,7 @@ class MarianConfig(PretrainedConfig):
def __init__(
self,
vocab_size=50265,
vocab_size=58101,
decoder_vocab_size=None,
max_position_embeddings=1024,
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