hparams : add SWA rope parameters (#12374)

ggml-ci
This commit is contained in:
Georgi Gerganov
2025-03-14 09:03:24 +02:00
committed by GitHub
parent 84d5475541
commit 081bee8c64
5 changed files with 26 additions and 20 deletions

View File

@ -70,7 +70,7 @@ uint32_t llama_hparams::n_embd_v_s() const {
return ssm_d_state * ssm_d_inner;
}
bool llama_hparams::is_sliding(uint32_t il) const {
bool llama_hparams::is_swa(uint32_t il) const {
if (il < n_layer) {
return n_swa > 0 && n_swa_pattern > 0 && il % n_swa_pattern < (n_swa_pattern - 1);
}