Clean special token init in modeling_....py (#3264)
* make style * fix conflicts
This commit is contained in:
committed by
GitHub
parent
8becb73293
commit
95e00d0808
@@ -41,9 +41,6 @@ class BartConfig(PretrainedConfig):
|
||||
activation_dropout=0.0,
|
||||
activation_function="gelu",
|
||||
vocab_size=50265,
|
||||
bos_token_id=0,
|
||||
pad_token_id=1,
|
||||
eos_token_ids=[2],
|
||||
d_model=1024,
|
||||
encoder_ffn_dim=4096,
|
||||
encoder_layers=12,
|
||||
@@ -61,6 +58,9 @@ class BartConfig(PretrainedConfig):
|
||||
output_past=False,
|
||||
num_labels=3,
|
||||
is_encoder_decoder=True,
|
||||
pad_token_id=1,
|
||||
bos_token_id=0,
|
||||
eos_token_id=2,
|
||||
**common_kwargs
|
||||
):
|
||||
r"""
|
||||
@@ -74,7 +74,7 @@ class BartConfig(PretrainedConfig):
|
||||
output_past=output_past,
|
||||
pad_token_id=pad_token_id,
|
||||
bos_token_id=bos_token_id,
|
||||
eos_token_ids=eos_token_ids,
|
||||
eos_token_id=eos_token_id,
|
||||
is_encoder_decoder=is_encoder_decoder,
|
||||
**common_kwargs,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user