EOS token for decode?

Is there a way to tell the audio decoder that a batch is finished? We have a larger batch of audio of varying length as indicated by an EOS token `[batch_size, codebooks, sample_length]` 

Currently the decoder just generates a continuous beep sound beyond the end of valid data.
 