From c59db4e0653f9daa7ceac162c90244e9daaafdc0 Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Thu, 15 Jun 2023 09:59:52 +0100 Subject: [PATCH] typo --- src/brevitas/graph/gptq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py index 6f598a888..45295f14a 100644 --- 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):