Added 'pad_token_id'. (#5)
Browse files- Added 'pad_token_id'. (5a0ba6d26ddd59f64e527d50f1311a52570e169a)
Co-authored-by: Petrov <sirorezka@users.noreply.huggingface.co>
- 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
|