Getting the error: "ValueError: The following model_kwargs are not used by the model:....."

[Pranav110500]

@XiangLi1999 Hello there, I am trying to run your code as it is with the commands provided with no changes whatsoever, however, I am still getting the following error.
"ValueError: The following model_kwargs are not used by the model: ['min_prob', 'student_lm', 'teacher_student', 'model_kwargs_student', 'st_coef', 'tokenizer', 'student_min_prob', 'student_temperature', 'use_cap_student', 'use_switch'] (note: typos in the generate arguments will also show up in this list)"

Can you help me with this? Is there any specific way to rectify this or am I supposed to simply drop these arguments from the parts where they are coded?