Train smaller language models with larger teachers.
Guides training smaller language models to imitate larger teachers using distillation losses and teacher-generated responses.
When to use it
Use when compressing language models, transferring teacher capabilities, reducing inference costs, or specializing smaller models.
Give it a language-model distillation task; it provides training code that trains and saves a smaller student model.
The training examples run as Python code.
The basic quick start places the student model on the CUDA device cuda:0.
This skill
meta-llama/Llama-2-70b-hf
Loads the teacher model
meta-llama/Llama-2-7b-hf
Loads the student model