Beam search is broken

[dreamgonfly]

To do beam search, I set cond_scale=1.0, turned off do_sample, and set num_beams > 1, then called model.generate().

However, I got the following error.

TypeError: reshape total size must be unchanged, got new_sizes (1, 2) for shape (12,).

Can you help me solve this issue?