mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-07-11 05:30:20 +00:00
add eagle3 support for Qwen3 series models
This commit is contained in:
@@ -21,6 +21,17 @@ llm_build_qwen3::llm_build_qwen3(const llama_model & model, const llm_graph_para
|
||||
for (int il = 0; il < n_layer; ++il) {
|
||||
ggml_tensor * inpSA = inpL;
|
||||
|
||||
// EAGLE3: Extract intermediate layer features from target model at layer INPUT
|
||||
if (eagle3 && cparams.eagle3_extract_enabled && !eagle3->extract_layer_indices.empty()) {
|
||||
static const char * eagle3_extract_names[] = {"eagle3_extract_0", "eagle3_extract_1", "eagle3_extract_2"};
|
||||
for (size_t i = 0; i < eagle3->extract_layer_indices.size() && i < 3; ++i) {
|
||||
if (eagle3->extract_layer_indices[i] == il) {
|
||||
cb(inpL, eagle3_extract_names[i], il);
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// norm
|
||||
cur = build_norm(inpL,
|
||||
model.layers[il].attn_norm, NULL,
|
||||
|
||||
Reference in New Issue
Block a user