fix issue with logit processor in beam search in Flax
Attach a file by drag & drop or click to upload
