zshanwei

hi @sacmehta Coremltools version==5.2.0. MacBook system version==12.4 the mobilevit-v2 model produced by example command is as follow. [MobileViTv2.mlmodel.zip](https://github.com/apple/ml-cvnets/files/9016909/MobileViTv2.mlmodel.zip)

> The converted graph looks good to me. I am not sure about the error that you are getting and I think it would be better if you post this...

Hi @sacmehta By analyzing the model, i have found that the problem is caused by dim value in torch.sum at https://github.com/apple/ml-cvnets/blob/main/cvnets/layers/linear_attention.py#L157 If dim is -1, the converted mlmodel will fail...