Merge pull request #3 from dcampora/elvis/clean-up-eagle

Clean up eagle

Author: dcampora

python/sglang/srt/configs/model_config.py

@@ -338,7 +338,6 @@ def _derive_model_shapes(self):
             or "DotsVLMForCausalLM" in self.hf_config.architectures
             or "MistralLarge3ForCausalLM" in self.hf_config.architectures
             or "PixtralForConditionalGeneration" in self.hf_config.architectures
-            or "MistralLarge3ForCausalLMEagle" in self.hf_config.architectures
         ):
             self.head_dim = 256
             self.attention_arch = AttentionArch.MLA

python/sglang/srt/models/mistral_large_3_eagle.py

@@ -70,16 +70,13 @@ def adapt_config_dict(
             "encoder_args"
         )
     )
-    is_eagle = "eagle" in model
 
     assert not (is_vision and is_audio), "Vision and audio are mutually exclusive"
 
     if is_vision:
         config_dict = _remap_mistral_vision_args(config_dict)
     if is_audio:
         config_dict = _remap_mistral_audio_args(config_dict)
-    if is_eagle:
-        config_dict = _remap_mistral_eagle_args(config_dict)
 
     config = PretrainedConfig.from_dict(config_dict)
 
@@ -211,11 +208,6 @@ def _remap_mistral_audio_args(config: dict) -> dict:
     return config
 
 
-def _remap_mistral_eagle_args(config: dict) -> dict:
-    config["architectures"] = ["MistralLarge3ForCausalLMEagle"]
-    return config
-
-
 def _remap_moe_args(config: dict) -> dict:
     moe_config_map = {
         "route_every_n": "moe_layer_freq",