Sagemaker Neo compilation fails

I have a PyTorch model that I've saved following https://docs.aws.amazon.com/sagemaker/latest/dg/neo-compilation-preparing-model.html into a .tar.gz file I uploaded it to S3, and then tried to compile it.

It fails with the error: ClientError: InputConfiguration: Framework cannot load PyTorch model. [enforce fail at inline_container.cc:222] . file not found: neo/version

My target platform is x86_64. Other settings are as in the screenshot. How do I fix this? thanks