fix scheduler

3 years ago · d0a754c1a1
parent e3a85872d0
commit d0a754c1a1
1 changed files with 1 additions and 1 deletions
--- a/swift/StableDiffusion/pipeline/Scheduler.swift
+++ b/swift/StableDiffusion/pipeline/Scheduler.swift
@ -179,7 +179,7 @@ public final class PNDMScheduler: Scheduler {
        timeSteps.append(contentsOf: forwardSteps.dropLast(1))
        timeSteps.append(timeSteps.last!)
        timeSteps.append(forwardSteps.last!)
-        timeSteps.reverse()
+        // do no revers timeSteps, this is now done in `calculateTimesteps` function
        self.timeSteps = timeSteps
        self.counter = 0