From 4aa84acf3a72b9cc07af931504a77ca9062a1238 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Wed, 14 Dec 2022 21:03:43 +0100 Subject: [PATCH] Remove additional extra space --- swift/StableDiffusion/pipeline/Scheduler.swift | 2 -- 1 file changed, 2 deletions(-) diff --git a/swift/StableDiffusion/pipeline/Scheduler.swift b/swift/StableDiffusion/pipeline/Scheduler.swift index 2f8202c..9e3ba8a 100644 --- a/swift/StableDiffusion/pipeline/Scheduler.swift +++ b/swift/StableDiffusion/pipeline/Scheduler.swift @@ -122,14 +122,12 @@ public final class PNDMScheduler: Scheduler { case .scaledLinear: self.betas = linspace(pow(betaStart, 0.5), pow(betaEnd, 0.5), trainStepCount).map({ $0 * $0 }) } - self.alphas = betas.map({ 1.0 - $0 }) var alphasCumProd = self.alphas for i in 1..