transformers

fix series of bugs for model parallel beam search

#46280

Merged