File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
vllm/transformers_utils/configs/speculators Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -22,6 +22,7 @@ def update_eagle3(config_dict: dict, pre_trained_config: dict) -> None:
2222 - draft_vocab_size: Size of the draft model's vocabulary
2323 - target_hidden_size: Hidden size of the target model
2424 - norm_before_residual: Whether to apply norm before residual connection
25+ - norm_before_fc: Whether to apply RMSNorm before the fc projection
2526 - eagle_aux_hidden_state_layer_ids: List of layer indices from the base
2627 model to use as auxiliary inputs for the Eagle3 drafter. These layers
2728 provide intermediate hidden states that help the drafter make better
@@ -34,6 +35,7 @@ def update_eagle3(config_dict: dict, pre_trained_config: dict) -> None:
3435 pre_trained_config ["norm_before_residual" ] = config_dict .get (
3536 "norm_before_residual" , True
3637 )
38+ pre_trained_config ["norm_before_fc" ] = config_dict .get ("norm_before_fc" , False )
3739 pre_trained_config ["architectures" ] = ["Eagle3LlamaForCausalLM" ]
3840 if config_dict .get ("eagle_aux_hidden_state_layer_ids" ):
3941 pre_trained_config ["eagle_aux_hidden_state_layer_ids" ] = config_dict [
You can’t perform that action at this time.
0 commit comments