mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-24 14:57:56 +08:00
Fix issue on old torch.
This commit is contained in:
parent
720a954e00
commit
554e371cad
@ -49,6 +49,7 @@ if RMSNorm is None:
|
|||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
self.register_parameter("weight", None)
|
self.register_parameter("weight", None)
|
||||||
|
self.bias = None
|
||||||
|
|
||||||
def forward(self, x):
|
def forward(self, x):
|
||||||
return rms_norm(x, self.weight, self.eps)
|
return rms_norm(x, self.weight, self.eps)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user