Report actual number of steps.

1 year ago · 2e05ad1386
parent 73927da7f9
commit 2e05ad1386
1 changed files with 1 additions and 1 deletions
--- a/swift/StableDiffusion/pipeline/StableDiffusionPipeline.swift
+++ b/swift/StableDiffusion/pipeline/StableDiffusionPipeline.swift
@ -219,7 +219,7 @@ public struct StableDiffusionPipeline: ResourceManaging {
                pipeline: self,
                prompt: prompt,
                step: step,
-                stepCount: stepCount,
+                stepCount: timeSteps.count,
                currentLatentSamples: latents,
                isSafetyEnabled: canSafetyCheck && !disableSafety
            )