How to add the original student model loss to the model distillation demo

Hello, I have tried to use the distillation script you provided（examples/llm_distill/main.py） to distill the LLM. However, I found that only the soft label is used in the script.
I would like to ask how to modify the code so that it can be combined with the original loss (hard label) of the student model. 
It would be of great help to me if you could reply! Thank you.