Fix eos_token_id to align with vocabulary of this model

#6
by mike-ravkine - opened
Files changed (1) hide show
  1. configuration_RW.py +1 -1
configuration_RW.py CHANGED
@@ -38,7 +38,7 @@ class RWConfig(PretrainedConfig):
38
  initializer_range=0.02,
39
  use_cache=True,
40
  bos_token_id=1,
41
- eos_token_id=2,
42
  apply_residual_connection_post_layernorm=False,
43
  hidden_dropout=0.0,
44
  attention_dropout=0.0,
 
38
  initializer_range=0.02,
39
  use_cache=True,
40
  bos_token_id=1,
41
+ eos_token_id=11,
42
  apply_residual_connection_post_layernorm=False,
43
  hidden_dropout=0.0,
44
  attention_dropout=0.0,