diff --git a/fs/ggml/ggml.go b/fs/ggml/ggml.go index 53d1543c2..c835cb32b 100644 --- a/fs/ggml/ggml.go +++ b/fs/ggml/ggml.go @@ -874,7 +874,7 @@ func (f GGML) SupportsFlashAttention() bool { return true } - if slices.Contains([]string{"gemma2"}, arch) { + if slices.Contains([]string{"gemma2", "grok"}, arch) { return false }