diff --git a/src/speculators/convert/eagle/eagle3_converter.py b/src/speculators/convert/eagle/eagle3_converter.py index 360383370..f66b6e3f6 100644 --- a/src/speculators/convert/eagle/eagle3_converter.py +++ b/src/speculators/convert/eagle/eagle3_converter.py @@ -167,6 +167,7 @@ def _create_transformer_config_from_eagle( head_dim=eagle_config.get("head_dim"), **build_llama_config_rope_kwargs( rope_theta=eagle_config.get("rope_theta", 10000.0), + rope_scaling=eagle_config.get("rope_scaling"), ), **build_llama_config_dtype_kwarg(eagle_config.get("torch_dtype")), )