]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
Change RMSNorm eps to 1e-6 (#173)
authorGeorgi Gerganov <redacted>
Sun, 19 Mar 2023 15:30:00 +0000 (17:30 +0200)
committerGeorgi Gerganov <redacted>
Sun, 19 Mar 2023 15:30:00 +0000 (17:30 +0200)
I think this is what is used in the Python code

ggml.c

diff --git a/ggml.c b/ggml.c
index 4fb83adbdb2c8c83450fa48d25e9a0f65fac911f..4813f74c895c950cd5c75309418dbcaf537893fe 100644 (file)
--- a/ggml.c
+++ b/ggml.c
@@ -5556,7 +5556,7 @@ static void ggml_compute_forward_rms_norm_f32(
     const size_t nb2 = dst->nb[2];
     const size_t nb3 = dst->nb[3];
 
-    const ggml_float eps = 1e-5f; // TODO: make this a parameter
+    const ggml_float eps = 1e-6f; // TODO: make this a parameter
 
     // TODO: optimize
     for (int i03 = 0; i03 < ne03; i03++) {
@@ -5572,7 +5572,7 @@ static void ggml_compute_forward_rms_norm_f32(
                 mean /= ne00;
 
                 float * y = (float *) ((char *) dst->data + i01*nb1 + i02*nb2 + i03*nb3);
-                
+
                 memcpy(y, x, ne00 * sizeof(float));
                 // for (int i00 = 0; i00 < ne00; i00++) {
                 //     y[i00] = x[i00];