llama : disable context-shift for DeepSeek v2 (#8501)

This commit is contained in:
Georgi Gerganov 2024-07-17 10:32:59 +03:00 committed by GitHub
parent 5e116e8dd5
commit d65a8361fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -15133,6 +15133,10 @@ static void llama_kv_cache_update_internal(struct llama_context & lctx) {
// apply K-shift if needed
if (lctx.model.hparams.rope_type != LLAMA_ROPE_TYPE_NONE && lctx.kv_self.has_shift) {
if (lctx.model.arch == LLM_ARCH_DEEPSEEK2) { // not supported due to MLA
GGML_ASSERT(false && "Deepseek2 does not support K-shift");
}
{
ggml_backend_sched_reset(lctx.sched);