Core ML. Style transfer, maximum size of model input/output

I'm playing with different input/output sizes of my model and increasing it at some point have the following error in my iPhone 7:
exception=Espresso exception: "Context exceeded": Cannot create 2147483648-byte metal heap. status=-13
But it works fine in simulator.
Is it possible to make a prediction to be sure that it will be enough memory to apply style transfer and don't have crash? And maybe there is some list with recommendations about input/output model sizes for different devices?
Thanks!
So if you can't process high res images then ML in iOS is really useless...
Core ML. Style transfer, maximum size of model input/output
 
 
Q