Added 'pad_token_id'.

#5
by sirorezka - opened
Files changed (1) hide show
  1. configuration_ouro.py +1 -0
configuration_ouro.py CHANGED
@@ -173,6 +173,7 @@ class OuroConfig(PretrainedConfig):
173
  early_exit_threshold=1.0,
174
  **kwargs,
175
  ):
 
176
  self.vocab_size = vocab_size
177
  self.max_position_embeddings = max_position_embeddings
178
  self.hidden_size = hidden_size
 
173
  early_exit_threshold=1.0,
174
  **kwargs,
175
  ):
176
+ self.pad_token_id = 2 # eom_token_id
177
  self.vocab_size = vocab_size
178
  self.max_position_embeddings = max_position_embeddings
179
  self.hidden_size = hidden_size