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)