Added 'pad_token_id'.
#5
by
sirorezka - opened
- 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
|