1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-03 15:39:54 +08:00

Fix issue on old torch.

This commit is contained in:
comfyanonymous
2025-04-16 04:53:56 -04:00
parent 9ad792f927
commit b4dc03ad76

View File

@@ -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)