diff --git a/coremltools/converters/mil/frontend/torch/ops.py b/coremltools/converters/mil/frontend/torch/ops.py index 6c69c94ad..8f0215b4f 100644 --- a/coremltools/converters/mil/frontend/torch/ops.py +++ b/coremltools/converters/mil/frontend/torch/ops.py @@ -5944,7 +5944,7 @@ def _translate_torch_args(dim) -> Var: context.add(res, torch_name=node.name) -@register_torch_op +@register_torch_op(torch_alias=["unbind_copy"]) def unbind(context, node): def _parse_positional_args(context, node) -> Tuple[Var]: inputs = _get_inputs(context, node, expected=(1, 2))