mtp init -2

This commit is contained in:
Concedo 2026-06-21 10:21:48 +08:00
parent a5019767c3
commit f202c0a457
2 changed files with 2 additions and 2 deletions

View file

@ -893,7 +893,7 @@ static void mtp_decoding_setup(llama_model * main_model, llama_context * main_ct
}
draft_is_mtp = true;
speculative_state_setup(main_ctx, mtp_ctx_params, -1, COMMON_SPECULATIVE_TYPE_DRAFT_MTP);
speculative_state_setup(main_ctx, mtp_ctx_params, -2, COMMON_SPECULATIVE_TYPE_DRAFT_MTP);
}
//loads a model for speculative decoding.

View file

@ -1,6 +1,6 @@
numpy>=1.24.4
sentencepiece>=0.1.98
transformers>=4.34.0
transformers[torch]>=4.34.0
gguf>=0.1.0
customtkinter>=5.1.0
protobuf>=4.21.0