diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py index b48cde64d..ab1384654 100644 --- a/src/brevitas/graph/gptq.py +++ b/src/brevitas/graph/gptq.py @@ -296,4 +296,3 @@ def initialize_module_optimizer( len_parallel_layers=len_parallel_layers, create_weight_orig=create_weight_orig, num_blocks=self.num_blocks) -