src/transformers/models/speech_to_text/modeling_tf_speech_to_text.py/TFSpeech2TextDecoder/set_embed_tokens
class TFSpeech2TextDecoder:    def set_embed_tokens(self, embed_tokens):
        self.embed_tokens = embed_tokens