Unverified Commit 4f9e71df authored by Yichen Yan's avatar Yichen Yan Committed by GitHub
Browse files

Remove duplicated code (#10545)


Co-authored-by: default avatarjiapingW <56055330+jiapingW@users.noreply.github.com>
parent 541551ce
...@@ -484,10 +484,6 @@ class Qwen2ForCausalLM(nn.Module): ...@@ -484,10 +484,6 @@ class Qwen2ForCausalLM(nn.Module):
if self.capture_aux_hidden_states: if self.capture_aux_hidden_states:
hidden_states, aux_hidden_states = 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 self.pp_group.is_last_rank:
if not get_embedding: if not get_embedding:
return self.logits_processor( return self.logits_processor(
......
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