small docstring changes and getting rid of unused functions/variables

Signed-off-by: Govind Ramnarayan <105831528+govind-ramnarayan@users.noreply.github.com>

Author: govind-ramnarayan

tensorrt_llm/_torch/auto_deploy/models/custom/modeling_eagle.py

@@ -15,16 +15,16 @@
 
 """Eagle3 model implementation for AutoDeploy.
 
-Eagle is a speculative decoding draft model that predicts next tokens based on
+Eagle3 is a speculative decoding draft model that predicts next tokens based on
 hidden states from a target model (e.g., Llama-3.1-8B-Instruct).
 
 This implementation:
-- Defines EagleConfig extending LlamaConfig with model_type="eagle3"
-- Wraps EagleModel in a HuggingFace-compatible interface
+- Defines Eagle3Config with model_type="eagle3"
+- Wraps Eagle3Model in a HuggingFace-compatible interface
 - Registers with AutoDeploy's custom model mechanism
 
-Note: Eagle uses the same tokenizer as its target model (Llama), so when using
-this model, you must explicitly specify the tokenizer path pointing to the
+Note: Eagle3 uses the same tokenizer as its target model (e.g., Llama), so when
+using this model, you must explicitly specify the tokenizer path pointing to the
 target model.
 """
 
@@ -317,13 +317,6 @@ class Eagle3ModelForCausalLM(PreTrainedModel):
     def __init__(self, config):
         super().__init__(config)
         self.model = Eagle3Model(config)
-        self._hidden_size = config.hidden_size
-        self._dtype = config.dtype
-
-    def _init_weights(self, module):
-        """Initialize weights - called by PreTrainedModel."""
-        # Default initialization - weights will be loaded from checkpoint
-        pass
 
     def forward(
         self,
@@ -419,6 +412,11 @@ class MockEagle3ModelForCausalLM(Eagle3ModelForCausalLM):
 
     config_class = MockEagle3Config
 
+    def __init__(self, config):
+        super().__init__(config)
+        self._hidden_size = config.hidden_size
+        self._dtype = config.dtype
+
     def forward(self, input_ids, **kwargs):
         # Inject mock hidden states if not provided
         if "target_hidden_states" not in kwargs: