ridger sirorezka commited on
Commit
bed729a
·
1 Parent(s): d917c94

Added 'pad_token_id'. (#5)

Browse files

- Added 'pad_token_id'. (5a0ba6d26ddd59f64e527d50f1311a52570e169a)


Co-authored-by: Petrov <sirorezka@users.noreply.huggingface.co>

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