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__(
self,
normalized_shape,
eps=None,
eps=1e-6,
elementwise_affine=True,
device=None,
dtype=None,