diff --git a/ops.py b/ops.py index 2d67e23..88ca894 100644 --- a/ops.py +++ b/ops.py @@ -76,7 +76,7 @@ class GGMLLayer(torch.nn.Module): self.to(x.device) weight, bias = self.get_weights(x.dtype) - x = self._forward_operation(x, weight, bias) + x = self._forward_operation(x, weight, bias=bias) del weight, bias if device: