default n_yarn_orig_ctx to n_ctx_train

This commit is contained in:
Cebtenzzre 2023-10-08 16:51:51 -04:00
parent babf0e0c8f
commit 0050e1ecc6

View file

@ -2011,7 +2011,7 @@ static void llm_load_hparams(
GGUF_GET_KEY(ctx, hparams.rope_finetuned, gguf_get_val_bool, GGUF_TYPE_BOOL, false,
kv(LLM_KV_ROPE_SCALING_FINETUNED));
hparams.n_yarn_orig_ctx = 0;
hparams.n_yarn_orig_ctx = hparams.n_ctx_train;
GGUF_GET_KEY(ctx, hparams.n_yarn_orig_ctx, gguf_get_val_u32, GGUF_TYPE_UINT32, false,
kv(LLM_KV_ROPE_SCALING_ORIG_CTX_LEN));