During BERT decoding, past_key_values is used to accelerate calculation. Do we have a similar implementation?

[etrigger]

I did not find such a cached method using past_key_values in the SAT. Is it possible to add this?
Thanks.

[Sleepy_chord]

Yes, but more simpler.
You can just do this model.add_mixin('auto-regressive', CachedAutoregressiveMixin()). You don't need to consider past_key_values when implementing model (In most cases), can this mixin and filling_sequence (autoregressive api) will save cache for it.

example see llama inference example