Open victor-psiori opened 3 years ago
Fixed the order of arguments for the _update_mems() function of MemTransformerLM class.
_update_mems()
MemTransformerLM
hids, mems, mlen, qlen
hids, mems, qlen mlen
qlen
mlen
Cheers, victor
Fixed the order of arguments for the
_update_mems()
function ofMemTransformerLM
class.hids, mems, mlen, qlen
whereas class function definition expects in order:hids, mems, qlen mlen
.qlen
andmlen
are added together for calculating context token length.Cheers, victor