"vscode:/vscode.git/clone" did not exist on "61de3ef74b9cd5594e1d79ba573258adf481382c"
Unverified Commit 60c872d4 authored by Chen Zhang's avatar Chen Zhang Committed by GitHub
Browse files

[Doc] Fix small typo in Transformers fallback (#14791)


Signed-off-by: default avatarChen Zhang <zhangch99@outlook.com>
parent 3fb17d26
......@@ -101,7 +101,7 @@ class MyAttention(nn.Module):
def forward(self, hidden_states, **kwargs): # <- kwargs are required
...
attention_interface = attention_interface = ALL_ATTENTION_FUNCTIONS[self.config._attn_implementation]
attention_interface = ALL_ATTENTION_FUNCTIONS[self.config._attn_implementation]
attn_output, attn_weights = attention_interface(
self,
query_states,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment