transformers

fix: loss computation after embeddings resize - mllama

#36840

Merged