src/transformers/models/jukebox/modeling_jukebox.py/JukeboxAttention/del_cache
class JukeboxAttention:    def del_cache(self):
        self.sample_t = 0
        if "key" in self.cache:
            del self.cache["key"]
        if "value" in self.cache:
            del self.cache["value"]
        self.cache = {}