transformers

fix issue with logit processor during beam search in Flax

#29636

Merged