Refine naming (#8868)

This commit is contained in:
Ke Bao
2025-08-06 21:37:02 +08:00
committed by GitHub
parent 1bd5316873
commit 399e7ec8b3
4 changed files with 30 additions and 30 deletions

View File

@@ -301,7 +301,7 @@ class GptOssAttention(nn.Module):
hidden_states, forward_batch, inner_state = intermediate_state
if inner_state is None:
return hidden_states
attn_output = self.attn(*inner_state, sk=self.sinks)
attn_output = self.attn(*inner_state, sinks=self.sinks)
output, _ = self.o_proj(attn_output)
return output