diff --git a/python/sglang/srt/models/qwen2.py b/python/sglang/srt/models/qwen2.py index b3d5fb9ad..256caee9c 100644 --- a/python/sglang/srt/models/qwen2.py +++ b/python/sglang/srt/models/qwen2.py @@ -484,10 +484,6 @@ class Qwen2ForCausalLM(nn.Module): if self.capture_aux_hidden_states: hidden_states, aux_hidden_states = hidden_states - aux_hidden_states = None - if self.capture_aux_hidden_states: - hidden_states, aux_hidden_states = hidden_states - if self.pp_group.is_last_rank: if not get_embedding: return self.logits_processor(