Refine naming (#8868)
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user