reset scheduler if default otherwise it will persist the old one

This commit is contained in:
Concedo
2025-10-19 19:53:40 +08:00
parent 7d20e6bdb3
commit d9e898afe0
2 changed files with 33 additions and 27 deletions
+3 -1
View File
@@ -948,7 +948,9 @@ public:
denoiser->scheduler = std::make_shared<SmoothStepSchedule>();
break;
case DEFAULT:
// Don't touch anything.
// Reset back to discrete
LOG_INFO("running with discrete scheduler");
denoiser->scheduler = std::make_shared<DiscreteSchedule>();
break;
default:
LOG_ERROR("Unknown scheduler %i", scheduler);