Fix comment.

pull/59/head
Pedro Cuenca 3 years ago
parent a44a91095d
commit 4a64fc1ff0

@ -37,7 +37,7 @@ public final class DPMSolverMultistepScheduler: Scheduler {
// Stores solverOrder (2) items // Stores solverOrder (2) items
private(set) var modelOutputs: [MLShapedArray<Float32>] = [] private(set) var modelOutputs: [MLShapedArray<Float32>] = []
/// Create a scheduler that uses a pseudo linear multi-step (PLMS) method /// Create a scheduler that uses a second order DPM-Solver++ algorithm.
/// ///
/// - Parameters: /// - Parameters:
/// - stepCount: Number of inference steps to schedule /// - stepCount: Number of inference steps to schedule

Loading…
Cancel
Save