Merge pull request #838 from gongel/feat_beam_search_v2

[Feat] Add Beam Search v2

Author: gongenlei

paddlenlp/ops/faster_transformer/transformer/faster_transformer.py

@@ -344,7 +344,8 @@ def __init__(self,
                     eos_id=eos_id,
                     beam_size=beam_size,
                     max_out_len=max_out_len,
-                    output_time_major=self.output_time_major)
+                    output_time_major=self.output_time_major,
+                    **kwargs)
         else:
             self.transformer = InferTransformerModel(
                 src_vocab_size=src_vocab_size,
@@ -361,7 +362,8 @@ def __init__(self,
                 eos_id=eos_id,
                 beam_size=beam_size,
                 max_out_len=max_out_len,
-                output_time_major=self.output_time_major)
+                output_time_major=self.output_time_major,
+                **kwargs)
 
     def forward(self, src_word):
         r"""