src/transformers/models/jukebox/modeling_jukebox.py/JukeboxLayerStack/del_cache
class JukeboxLayerStack:    def del_cache(self):
        for attn_layer in self._attn_mods:
            attn_layer.attn.del_cache()