From 258742c369f021c2e5e74fd5bcf2a45a6a5d30d1 Mon Sep 17 00:00:00 2001 From: City <125218114+city96@users.noreply.github.com> Date: Mon, 26 Aug 2024 17:49:18 +0200 Subject: [PATCH] Fix Conv2D with bias --- ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: