Is it a typo?

[dwangF0]

In the main branch, line 671 of this file ./audiolm_pytorch/audiolm_pytorch.py,

line 671 "return logits, new_kv_cache"

Is it supposed to be "return logits, kv_cache"? I searched the whole forward function, and cannot find new_kv_cache, but only kv_cache.

Does anyone have any idea of this?

[lucidrains]

oops yea, should be fixed

[dwangF0]

no worry, thanks for the quick response!