Skip to content

Commit

Permalink
fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
ArthurZucker committed Dec 7, 2023
1 parent e370dfd commit 1ed661c
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion src/transformers/models/chatglm/modeling_chatglm.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 1ed661c

Please sign in to comment.