fix(generation): beam sample when num_beams * vocab_size exceeds multinomial limit#45251

balgaly wants to merge 2 commits intohuggingface:mainhuggingface/transformers:mainfrom

balgaly:fix/beam-sample-multinomial-flat-dim-limitbalgaly/transformers:fix/beam-sample-multinomial-flat-dim-limitCopy head branch name to clipboard