Skip to content

GitLab

Unverified Commit d42975c6 authored by Lzhang-hub's avatar Lzhang-hub Committed by GitHub
Browse files

Remove duplicate code in qwen2 model (#10540)

parent adc24a3a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
python/sglang/srt/models/qwen2.py
View file @ d42975c6
...@@ -454,9 +454,6 @@ class Qwen2ForCausalLM(nn.Module): ...@@ -454,9 +454,6 @@ class Qwen2ForCausalLM(nn.Module):
# For EAGLE3 support # For EAGLE3 support
self.capture_aux_hidden_states = False self.capture_aux_hidden_states = False
# For EAGLE3 support
self.capture_aux_hidden_states = False
def get_input_embedding(self, input_ids: torch.Tensor) -> torch.Tensor: def get_input_embedding(self, input_ids: torch.Tensor) -> torch.Tensor:
return self.model.get_input_embedding(input_ids) return self.model.get_input_embedding(input_ids)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment