EDM2 and Autoguidance -- Official PyTorch implementation - View it on GitHub