Remove duplicate code in qwen2 model (#10540)
This commit is contained in:
@@ -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)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user