From 080b5d0df4a36628d5ea33618e2a518c63ea2a3a Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Mon, 5 May 2025 01:46:04 +0300
Subject: [PATCH] Update rmsnorm.py (Ensuring eps is not None)

---
 comfy/rmsnorm.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/comfy/rmsnorm.py b/comfy/rmsnorm.py
index 9d82bee1a..bb2f6fa74 100644
--- a/comfy/rmsnorm.py
+++ b/comfy/rmsnorm.py
@@ -12,6 +12,10 @@ except:
 
 
 def rms_norm(x, weight=None, eps=1e-6):
+    
+    if eps is None: # Ensure eps is not None
+        eps = 1e-6
+        
     if rms_norm_torch is not None and not (torch.jit.is_tracing() or torch.jit.is_scripting()):
         if weight is None:
             return rms_norm_torch(x, (x.shape[-1],), eps=eps)
@@ -30,7 +34,7 @@ if RMSNorm is None:
         def __init__(
             self,
             normalized_shape,
-            eps=None,
+            eps=1e-6,  # Changed default from None to 1e-6
             elementwise_affine=True,
             device=None,
             dtype=None,