nilla pipeline when using ToMe and ToMe + xFormers. GPU Resolution Batch size Vanilla ToMe ToMe + xFormers A100 512 10 6.88 5.26 (+23.55%) 4.69 (+31.83%)  768 10 OOM 14.71 11   8 OOM 11.56 8.84   4 OOM 5.98 4.66   2 4.99 3.24 (+35.07%) 2.1 (+37.88%)   1 3.29 2.24 (+31.91%) 2.03 (+38.3%)  1024 10 OOM OOM OOM   8 OOM OOM OOM   4 OOM 12.51 9.09   2 OOM 6.52 4.96   1 6.4 3.61 (+43.59%) 2.81 (+56.09%) V100 512 10 OOM 10.03 9.29   8 OOM 8.05 7.47   4