diff --git a/docs/source/models/supported_models.md b/docs/source/models/supported_models.md index bcbd7bf9..3d42d5f6 100644 --- a/docs/source/models/supported_models.md +++ b/docs/source/models/supported_models.md @@ -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,