Fix Conv2D with bias

2024-08-26 17:49:18 +02:00
parent a9f6359a23
commit 258742c369
1 changed files with 1 additions and 1 deletions
@@ -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: