Fix small typo in JambaDecoder
#24
by mber - opened
great job and wonderful implementation! just fixing a small typo in the comments of the shapes of the tensors, since the intermediate dimension is still in shape[1] in that line.
Thanks!
tomeras1 changed pull request status to merged