clamp initial timestamp

swift/StableDiffusion/pipeline/AlphasCumprodCalculation.swift

@@ -22,7 +22,7 @@ public struct AlphasCumprodCalculation {
         strength: Float
     ) {
         let tEnc = Int(strength * Float(steps))
-        let initTimestep = timesteps - timesteps / steps * (steps - tEnc) + 1
+        let initTimestep = min(max(0, timesteps - timesteps / steps * (steps - tEnc) + 1), timesteps - 1)
         self.sqrtAlphasCumprod = alphasCumprod[initTimestep].squareRoot()
         self.sqrtOneMinusAlphasCumprod = (1 - alphasCumprod[initTimestep]).squareRoot()
     }

swift/StableDiffusion/pipeline/Encoder.swift

@@ -66,7 +66,6 @@ public struct Encoder {
             else {
                 throw Error.latentOutputNotValid
             }
-            print("output.shape: \(output.shape)")
             return MLShapedArray(output)
         }