mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-04-19 10:53:29 +00:00
Fix issue on old torch.
This commit is contained in:
parent
9ad792f927
commit
b4dc03ad76
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user