Fix small typo in JambaDecoder

#24

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.

AI21 org

Thanks!

tomeras1 changed pull request status to merged

Sign up or log in to comment