diff --git a/swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift b/swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift index 26f3854..d84a4ef 100644 --- a/swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift +++ b/swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift @@ -37,7 +37,7 @@ public final class DPMSolverMultistepScheduler: Scheduler { // Stores solverOrder (2) items private(set) var modelOutputs: [MLShapedArray] = [] - /// Create a scheduler that uses a pseudo linear multi-step (PLMS) method + /// Create a scheduler that uses a second order DPM-Solver++ algorithm. /// /// - Parameters: /// - stepCount: Number of inference steps to schedule