mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-06-10 13:32:08 +08:00
Fix issue with old pytorch RMSNorm. (#8095)
This commit is contained in:
parent
481732a0ed
commit
a814f2e8cc
@ -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,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user