Fix issue with old pytorch RMSNorm. (#8095)

This commit is contained in:
comfyanonymous 2025-05-13 04:54:28 -07:00 committed by GitHub
parent 481732a0ed
commit a814f2e8cc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -30,7 +30,7 @@ if RMSNorm is None:
def __init__( def __init__(
self, self,
normalized_shape, normalized_shape,
eps=None, eps=1e-6,
elementwise_affine=True, elementwise_affine=True,
device=None, device=None,
dtype=None, dtype=None,