Me at the end of last year: "Should I really go for the full 64GB on this M1 Pro? What could I possibly use this for? mmbml mumble... something about unified GPU... something Deep Learning, one day..."
Me now: "a red juicy apple floating in outer space, like a planet" --H 768 --W 768
Uses about 27GB. 1.81s/it.
Can't do 1024x1024 yet because of some hardcoded Metal issue (https://github.com/pytorch/pytorch/issues/84039.