mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-08-12 22:31:11 +04:00
fix sliding_window_pattern
This commit is contained in:
@@ -66,7 +66,12 @@ void llama_model_dflash::load_arch_hparams(llama_model_loader & ml) {
|
|||||||
// DFlash has a single rope, so the SWA rope == main rope.
|
// DFlash has a single rope, so the SWA rope == main rope.
|
||||||
if (ml.get_key(LLM_KV_ATTENTION_SLIDING_WINDOW, hparams.n_swa, false) && hparams.n_swa > 0) {
|
if (ml.get_key(LLM_KV_ATTENTION_SLIDING_WINDOW, hparams.n_swa, false) && hparams.n_swa > 0) {
|
||||||
hparams.swa_type = LLAMA_SWA_TYPE_STANDARD;
|
hparams.swa_type = LLAMA_SWA_TYPE_STANDARD;
|
||||||
ml.get_key_or_arr(LLM_KV_ATTENTION_SLIDING_WINDOW_PATTERN, hparams.is_swa_impl, hparams.n_layer());
|
uint32_t swa_period = 0;
|
||||||
|
if (ml.get_key_or_arr(LLM_KV_ATTENTION_SLIDING_WINDOW_PATTERN, swa_period, false)) {
|
||||||
|
hparams.set_swa_pattern(swa_period);
|
||||||
|
} else {
|
||||||
|
ml.get_key_or_arr(LLM_KV_ATTENTION_SLIDING_WINDOW_PATTERN, hparams.is_swa_impl, hparams.n_layer_all);
|
||||||
|
}
|
||||||
hparams.rope_freq_base_train_swa = hparams.rope_freq_base_train;
|
hparams.rope_freq_base_train_swa = hparams.rope_freq_base_train;
|
||||||
hparams.rope_freq_scale_train_swa = hparams.rope_freq_scale_train;
|
hparams.rope_freq_scale_train_swa = hparams.rope_freq_scale_train;
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user