The ONNX model's constant layer tensor dimensions size must be 1, 2, 3 or 4 Snapchat Lens Studio Error

Question

I exported my Pytorch model to ONNX to be a SnapML model but when I tried to load it into lens studio I get the error message "The ONNX model\'s constant layer tensor d