From 9f7f205f67f35ec0ea90caf856e9cb2774544f14 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 5 Mar 2024 18:27:11 +1100 Subject: [PATCH] Update rms_layernorm.py --- unsloth/kernels/rms_layernorm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/kernels/rms_layernorm.py b/unsloth/kernels/rms_layernorm.py index a341f3e154..cae9d30b21 100644 --- a/unsloth/kernels/rms_layernorm.py +++ b/unsloth/kernels/rms_layernorm.py @@ -161,6 +161,6 @@ pass def fast_rms_layernorm(layernorm, X, add_one = False): W = layernorm.weight eps = layernorm.variance_epsilon - out = Fast_RMS_Layernorm.apply(X, W, eps, add_one = add_one) + out = Fast_RMS_Layernorm.apply(X, W, eps, add_one) return out pass