From bd60010baaf31548111d0b123af22fcc5f621480 Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Wed, 14 Dec 2022 20:47:26 +0100 Subject: [PATCH] Remove blank space --- 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 6658b85..fc10bbc 100644 --- a/swift/StableDiffusion/pipeline/Scheduler.swift +++ b/swift/StableDiffusion/pipeline/Scheduler.swift @@ -115,7 +115,7 @@ public final class PNDMScheduler: Scheduler { ) { self.trainStepCount = trainStepCount self.inferenceStepCount = stepCount - + switch betaSchedule { case .linear: self.betas = linspace(betaStart, betaEnd, trainStepCount)