fix scheduler

1 year ago · 0a1b2b87b3
parent effc166c62
commit 0a1b2b87b3
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