kd13
/

nano-translate-v1

lstm_seq2seq_en_hi

text2text-generation

Model card Files Files and versions

kd13 commited on Mar 28

Commit

772cfe7

·

verified ·

1 Parent(s): f7cc378

Update modeling_lstm_seq2seq_en_hi.py

Files changed (1) hide show

modeling_lstm_seq2seq_en_hi.py +3 -1

modeling_lstm_seq2seq_en_hi.py CHANGED Viewed

@@ -215,4 +215,6 @@ class Seq2SeqHFModel(PreTrainedModel):
         hidden, cell = past_key_values
         hidden = hidden.index_select(1, beam_idx)
         cell = cell.index_select(1, beam_idx)
-        return (hidden, cell)

         hidden, cell = past_key_values
         hidden = hidden.index_select(1, beam_idx)
         cell = cell.index_select(1, beam_idx)
+        return (hidden, cell)
+Seq2SeqHFModel.register_for_auto_class("AutoModelForSeq2SeqLM")