Spaces:
Runtime error
Runtime error
fix: typo
Browse filesFormer-commit-id: 812d34f157e905ab47e1081b79cc0a80b37fd19b
seq2seq/run_seq2seq_flax.py
CHANGED
|
@@ -273,7 +273,7 @@ class CustomFlaxBartModule(FlaxBartModule):
|
|
| 273 |
def setup(self):
|
| 274 |
# check config is valid, otherwise set default values
|
| 275 |
self.config.vocab_size_output = getattr(self.config, 'vocab_size_output', OUTPUT_VOCAB_SIZE)
|
| 276 |
-
self.config.max_position_embeddings_decoder = getattr(self.config, '
|
| 277 |
|
| 278 |
# we keep shared to easily load pre-trained weights
|
| 279 |
self.shared = nn.Embed(
|
|
|
|
| 273 |
def setup(self):
|
| 274 |
# check config is valid, otherwise set default values
|
| 275 |
self.config.vocab_size_output = getattr(self.config, 'vocab_size_output', OUTPUT_VOCAB_SIZE)
|
| 276 |
+
self.config.max_position_embeddings_decoder = getattr(self.config, 'max_position_embeddings_decoder', OUTPUT_LENGTH)
|
| 277 |
|
| 278 |
# we keep shared to easily load pre-trained weights
|
| 279 |
self.shared = nn.Embed(
|