Fix beam search for mamba models (#46819)

Commit

25 days ago

Fix beam search for mamba models (#46819) * fix * fix * add proper exception * forgot one * small fix