feat: add exponential scheduler (#346)

* feat: added exponential scheduler

* updated README

* improved exponential formatting

---------

Co-authored-by: leejet <leejet714@gmail.com>
This commit is contained in:
Daniele
2024-08-27 16:13:35 +00:00
committed by GitHub
parent d00c94844d
commit dc0882cdc9
5 changed files with 29 additions and 2 deletions

View File

@@ -538,6 +538,10 @@ public:
LOG_INFO("running with Karras schedule");
denoiser->schedule = std::make_shared<KarrasSchedule>();
break;
case EXPONENTIAL:
LOG_INFO("running exponential schedule");
denoiser->schedule = std::make_shared<ExponentialSchedule>();
break;
case AYS:
LOG_INFO("Running with Align-Your-Steps schedule");
denoiser->schedule = std::make_shared<AYSSchedule>();