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
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:
/// - stepCount: Number of inference steps to schedule

Loading…
Cancel
Save