Report actual number of steps.

3 years ago · f816ed816f
parent d9563fb586
commit f816ed816f
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
            )