Unverified Commit 8f577dca authored by Chris McMaster's avatar Chris McMaster Committed by GitHub
Browse files

Fixed typo in error message (#27461)

"past key much have a shape" -> "past key must have a shape"
parent 7b998cab
...@@ -372,7 +372,7 @@ class MistralFlashAttention2(MistralAttention): ...@@ -372,7 +372,7 @@ class MistralFlashAttention2(MistralAttention):
if past_key.shape[-2] != self.config.sliding_window - 1: if past_key.shape[-2] != self.config.sliding_window - 1:
raise ValueError( raise ValueError(
f"past key much have a shape of (`batch_size, num_heads, self.config.sliding_window-1, head_dim`), got" f"past key must have a shape of (`batch_size, num_heads, self.config.sliding_window-1, head_dim`), got"
f" {past_key.shape}" f" {past_key.shape}"
) )
......
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