Skip to content

Commit a814f2e

Browse files
Fix issue with old pytorch RMSNorm. (Comfy-Org#8095)
1 parent 481732a commit a814f2e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/rmsnorm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ class RMSNorm(torch.nn.Module):
3030
def __init__(
3131
self,
3232
normalized_shape,
33-
eps=None,
33+
eps=1e-6,
3434
elementwise_affine=True,
3535
device=None,
3636
dtype=None,

0 commit comments

Comments
 (0)