From 1ed661c5f924103f4f4d0fc512e0224656515558 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Thu, 7 Dec 2023 12:20:52 +0100 Subject: [PATCH] fixup --- src/transformers/models/chatglm/modeling_chatglm.py | 1 - 1 file changed, 1 deletion(-) 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,