Update src/llama-model.cpp
Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>
This commit is contained in:
parent
4187cf5a94
commit
e3ac2ae531
|
|
@ -8122,7 +8122,7 @@ struct llm_build_modern_bert : public llm_graph_context {
|
||||||
cur = inpL;
|
cur = inpL;
|
||||||
|
|
||||||
cur = build_norm(cur,
|
cur = build_norm(cur,
|
||||||
model.output_norm_enc, NULL,
|
model.output_norm, NULL,
|
||||||
LLM_NORM, -1);
|
LLM_NORM, -1);
|
||||||
cb(cur, "final_norm_out", -1);
|
cb(cur, "final_norm_out", -1);
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue