Fix bug

ModelTC · Jun 4, 2024 · b645901 · b645901
1 parent b824375
commit b645901
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmc/compression/quantization/quant.py b/llmc/compression/quantization/quant.py
@@ -181,7 +181,7 @@ def reshape_tensor(self, tensor):
  else:
  t = tensor
  elif self.granularity == "per_head":
- t = tensor.reshape(self.heda_num, -1)
+ t = tensor.reshape(self.head_num, -1)
  else:
  t = tensor
  return t