yilian49

Fix for when draft model hidden dimension is different from target model hidden dimension

2

## Motivation The current draft model files (e.g. llama3_eagle.py) supports draft model config having `target_hidden_dimension` separate from `hidden_dimension`. However, in the training scripts, it loads the target model's embedding function...