Fix xLSTM train mode crash when return_last_states=False#47026

lcheng321 wants to merge 2 commits into

huggingface:mainhuggingface/transformers:mainfrom

lcheng321:fix-xlstm-train-return-last-stateslcheng321/transformers:fix-xlstm-train-return-last-statesCopy head branch name to clipboard