Fix (QuantLayer): make bias optional for QuantRNN and QuantLSTM

Xilinx · Feb 14, 2024 · 9675e4c · 9675e4c
1 parent d58dc90
commit 9675e4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/brevitas/nn/quant_rnn.py b/src/brevitas/nn/quant_rnn.py
@@ -882,7 +882,7 @@ def __init__(
             hidden_size: int,
             num_layers: int = 1,
             nonlinearity: str = 'tanh',
-            bias: bool = True,
+            bias: Optional[bool] = True,
             batch_first: bool = False,
             bidirectional: bool = False,
             weight_quant=Int8WeightPerTensorFloat,
@@ -921,7 +921,7 @@ def __init__(
             input_size: int,
             hidden_size: int,
             num_layers: int = 1,
-            bias: bool = True,
+            bias: Optional[bool] = True,
             batch_first: bool = False,
             bidirectional: bool = False,
             weight_quant=Int8WeightPerTensorFloat,