and the speed difference is there. torch 2 at 120 steps 1 image is done in 8 seconds, now with torch
and the speed difference is there. torch 2 at 120 steps 1 image is done in 8 seconds, now with torch: 1.13.1+cu117 10 seconds //// 14it/s vs 10-11it/s
