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:
self.register_parameter("weight", None)
self.bias = None
def forward(self, x):
return rms_norm(x, self.weight, self.eps)