fix(models): build spatial_shapes from Python ints under torch.compile

fix(models): build spatial_shapes from Python ints under torch.compile #3558