Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/models/supported_models.md
View file @ 60c872d4
......@@ -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