github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-08-11 21:02:12 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats 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 // embd norm
{ {
inpL = ggml_norm(ctx0, inpL, 1e-5f); inpL = ggml_norm(ctx0, inpL, 1e-12f);
inpL = ggml_add(ctx0, inpL = ggml_add(ctx0,
ggml_mul(ctx0, ggml_mul(ctx0,
@ -403,7 +403,7 @@ void bert_eval(
// attention norm // attention norm
{ {
cur = ggml_norm(ctx0, cur, 1e-5f); cur = ggml_norm(ctx0, cur, 1e-12f);
cur = ggml_add(ctx0, cur = ggml_add(ctx0,
ggml_mul(ctx0, ggml_mul(ctx0,
@ -429,7 +429,7 @@ void bert_eval(
// output norm // output norm
{ {
cur = ggml_norm(ctx0, cur, 1e-5f); cur = ggml_norm(ctx0, cur, 1e-12f);
cur = ggml_add(ctx0, cur = ggml_add(ctx0,
ggml_mul(ctx0, ggml_mul(ctx0,