vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix kq_scale for Baichuan

Browse source
This commit is contained in:
KerfuffleV2 2023-10-12 14:23:08 -06:00
parent 5a06711f64
commit ae31a9a0b6
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama.cpp
View file

@ -3502,6 +3502,17 @@ struct llm_build_baichuan_ctx : llm_build_llama_ctx {
: llm_build_llama_ctx(lctx, batch) : llm_build_llama_ctx(lctx, batch)
{} {}
struct ggml_tensor * build_kq_scale() {
// KQ_scale
ggml_tensor * tensor = ggml_new_tensor_1d(ctx0, GGML_TYPE_F32, 1);
ggml_set_name(tensor, "1/sqrt(n_embd_head)");
ggml_allocr_alloc(alloc, tensor);
if (!alloc_measure) {
ggml_set_f32(KQ_scale, 1.0f/sqrtf(float(n_embd)/n_head));
}
return tensor;
}
std::tuple<ggml_tensor *, ggml_tensor *> build_attn_block_kcur_qcur( std::tuple<ggml_tensor *, ggml_tensor *> build_attn_block_kcur_qcur(
const llama_layer & layer, const llama_layer & layer,
ggml_tensor * cur) { ggml_tensor * cur) {