To reproduce this run:
```
accelerate launch --multi_gpu --mixed_precision=fp16 --num_processes=8 run_distillation config_mistral.yaml
```