diff --git a/src/transformers/models/chatglm/modeling_chatglm.py b/src/transformers/models/chatglm/modeling_chatglm.py index ea4701bb3092ae..ec9a900d30ddc4 100644 --- a/src/transformers/models/chatglm/modeling_chatglm.py +++ b/src/transformers/models/chatglm/modeling_chatglm.py @@ -262,7 +262,6 @@ def forward(self, hidden_states): return hidden_states - def repeat_kv(hidden_states: torch.Tensor, n_rep: int) -> torch.Tensor: """ This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch,