[converter] fix inhashable keys for group_tensors_pass (#348)

alibaba · Aug 20, 2024 · 345c1cd · 345c1cd
1 parent 7cc3d2e
commit 345c1cd
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tinynn/converter/operators/optimize.py b/tinynn/converter/operators/optimize.py
@@ -928,12 +928,18 @@ def group_tensors_pass(self):
                 if tensor.quantization is None:
                     t_idx = (tensor.buffer.data, tensor.dtype, tensor.shape)
                 else:
+                    scale = tensor.quantization.scale
+                    zero_point = tensor.quantization.zero_point
+                    if isinstance(scale, list):
+                        scale = tuple(scale)
+                    if isinstance(zero_point, list):
+                        zero_point = tuple(zero_point)
                     t_idx = (
                         tensor.buffer.data,
                         tensor.dtype,
                         tensor.shape,
-                        tensor.quantization.scale,
-                        tensor.quantization.zero_point,
+                        scale,
+                        zero_point,
                         tensor.quantization.dim,
                     )