kd13 commited on
Commit
4e38472
·
verified ·
1 Parent(s): 772cfe7

Update configuration_lstm_seq2seq_en_hi.py

Browse files
configuration_lstm_seq2seq_en_hi.py CHANGED
@@ -15,6 +15,7 @@ class Seq2SeqConfig(PretrainedConfig):
15
  eos_token_id=3,
16
  decoder_start_token_id=2,
17
  is_encoder_decoder=True,
 
18
  **kwargs
19
  ):
20
  super().__init__(
@@ -23,6 +24,7 @@ class Seq2SeqConfig(PretrainedConfig):
23
  eos_token_id=eos_token_id,
24
  decoder_start_token_id=decoder_start_token_id,
25
  is_encoder_decoder=is_encoder_decoder,
 
26
  **kwargs
27
  )
28
  self.vocab_size = vocab_size
 
15
  eos_token_id=3,
16
  decoder_start_token_id=2,
17
  is_encoder_decoder=True,
18
+ architectures=None,
19
  **kwargs
20
  ):
21
  super().__init__(
 
24
  eos_token_id=eos_token_id,
25
  decoder_start_token_id=decoder_start_token_id,
26
  is_encoder_decoder=is_encoder_decoder,
27
+ architectures=architectures or ["Seq2SeqHFModel"],
28
  **kwargs
29
  )
30
  self.vocab_size = vocab_size