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

llama : prep ALiBi support for BERT models

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2024-04-23 17:24:28 +03:00
parent 78d363b0d4
commit 19e8982f51
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
2 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ggml.c
View file

@ -5476,6 +5476,7 @@ static struct ggml_tensor * ggml_soft_max_impl(
GGML_ASSERT(mask->type == GGML_TYPE_F16 || mask->type == GGML_TYPE_F32);
GGML_ASSERT(ggml_is_contiguous(mask));
GGML_ASSERT(ggml_is_matrix(mask));
GGML_ASSERT(mask->ne[0] == a->ne[0]);
GGML_ASSERT(mask->ne[1] >= a->ne[1]);
}