typo

Xilinx · Jun 15, 2023 · c59db4e · c59db4e
1 parent 6327b1a
commit c59db4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py
@@ -133,7 +133,7 @@ def __enter__(self):
             self.disable_quant_inference.disable_bias_quantization(
                 self.model, is_training=self.model.training)
 
-        self.num_layers = len(list_of_layers)
+        self.num_layers = len(dict_of_layers)
         return self
 
     def __exit__(self, type, value, traceback):