Skip to content

Commit ed91d96

Browse files
authored
[quantization] Tidy SensitivityCalibrator (#582)
1 parent 976b8a9 commit ed91d96

File tree

1 file changed

+0
-2
lines changed
  • tico/quantization/algorithm/gptq

1 file changed

+0
-2
lines changed

tico/quantization/algorithm/gptq/utils.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -152,13 +152,11 @@ def compute_sensitivity_info(self):
152152

153153
sensitivity = {}
154154
modules_to_process = {}
155-
name_of_module: dict[torch.nn.Module, str] = {}
156155

157156
for name, module in model.named_modules():
158157
for type in self.calibrated_types:
159158
if isinstance(module, type):
160159
modules_to_process[name] = module
161-
name_of_module[module] = name
162160
sensitivity[name] = torch.zeros_like(module.weight).cpu()
163161
break
164162

0 commit comments

Comments
 (0)