/SFT-PG

Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023

Primary LanguagePythonMIT LicenseMIT

Stargazers