src/transformers/models/whisper/modeling_flax_whisper.py/FlaxWhisperModule/setup
class FlaxWhisperModule:    def setup(self) -> None:
        self.encoder = FlaxWhisperEncoder(self.config, dtype=self.dtype)
        self.decoder = FlaxWhisperDecoder(self.config, dtype=self.dtype)