in models/context_model.py [0:0]
def _reset(self):
"""
reset buffers in each layer before animating a new sequence
"""
self.embedding.reset()
for layer in self.context_layers:
layer.reset()
self.logits.reset()