src/transformers/models/jukebox/modeling_jukebox.py/JukeboxPreTrainedModel/_init_weights
class JukeboxPreTrainedModel:    def _init_weights(self, module):
        if isinstance(module, JukeboxPrior) or isinstance(module, JukeboxVQVAE):
            module.apply(module._init_weights)