Fix issue with old pytorch RMSNorm.

This commit is contained in:
comfyanonymous 2025-05-13 07:53:49 -04:00 committed by GitHub
parent 481732a0ed
commit 3338f24f29
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,