From d0a754c1a1d2de9071ad8c703ad89bc8c8eb7e7a Mon Sep 17 00:00:00 2001 From: Timothy Kautz Date: Mon, 19 Dec 2022 01:23:31 +0900 Subject: [PATCH] fix scheduler --- swift/StableDiffusion/pipeline/Scheduler.swift | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swift/StableDiffusion/pipeline/Scheduler.swift b/swift/StableDiffusion/pipeline/Scheduler.swift index d0ed12e..cfaa71c 100644 --- 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