github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-09-08 11:58:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

bert: fix layer norm epsilon value (#1946)

Browse Source 
Signed-off-by: Jared Van Bortel <jared@nomic.ai>
This commit is contained in:
Jared Van Bortel
2024-02-26 13:09:01 -05:00
committed by GitHub
parent 7a23b23728
commit 007d469034
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
gpt4all-backend/bert.cpp
View File

@@ -343,7 +343,7 @@ void bert_eval(
// embd norm
{
inpL = ggml_norm(ctx0, inpL, 1e-5f);
inpL = ggml_norm(ctx0, inpL, 1e-12f);
inpL = ggml_add(ctx0,
ggml_mul(ctx0,
@@ -403,7 +403,7 @@ void bert_eval(
// attention norm
{
cur = ggml_norm(ctx0, cur, 1e-5f);
cur = ggml_norm(ctx0, cur, 1e-12f);
cur = ggml_add(ctx0,
ggml_mul(ctx0,
@@ -429,7 +429,7 @@ void bert_eval(
// output norm
{
cur = ggml_norm(ctx0, cur, 1e-5f);
cur = ggml_norm(ctx0, cur, 1e-12f);
cur = ggml_add(ctx0,
ggml_mul(ctx0,