self._generate_beam_search question

[githubzmt]

Thank you very much for providing such an excellent paper, but during the pre training process, we found that some functions were missing. The content in the picture.

[Tsukiaa]

Yes, I also have this problem!

[hao920293897]

Yes, I also have this problem!