    def set_beam_size(self, beam_size):
        """Used for effiecient beamable enc-dec attention"""
        self.beam_size = beam_size
Used for effiecient beamable enc-dec attention
fairseq/modules/multihead_attention.py