Skip to content

Commit

Permalink
remove metal check
Browse files Browse the repository at this point in the history
  • Loading branch information
JohannesGaessler committed Aug 24, 2024
1 parent 832c6ee commit 6e40804
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions src/llama.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -17534,14 +17534,6 @@ struct llama_context * llama_new_context_with_model(
params.flash_attn = false;
}

#ifdef GGML_USE_METAL
if (params.flash_attn && model->hparams.attn_soft_cap) {
LLAMA_LOG_WARN("%s: flash_attn is not compatible with attn_soft_cap - forcing off\n", __func__);
params.flash_attn = false;
}
#endif // GGML_USE_METAL


if (params.flash_attn && model->hparams.n_embd_head_k != model->hparams.n_embd_head_v) {
LLAMA_LOG_WARN("%s: flash_attn requires n_embd_head_k == n_embd_head_v - forcing off\n", __func__);
params.flash_attn = false;
Expand Down

0 comments on commit 6e40804

Please sign in to comment.