Qwen Student and teacher logits mismatch during KD, needs padding

[nouranali]

In this exact line, in kd_loss.py

[ebsmothers]

Hi @nouranali thanks for creating the issue! I left a comment on #2599, happy to continue the discussion there.