1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00

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

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,