mirrors / apple--ml-stable-diffusion

项目文件夹

文件

T

Guillermo Cique 1147e87b79 Simplify Encoder module (#145 )

* Simplify encoder conversion

* Add nextNormal to RandomSource

* Add addNoise method to Scheduler

* Update Encoder implementation

* Clean up Pipeline

* Remove AlphasCumprodCalculation

* Revert Encoder’s input shape size

* Make Encoder/Decoder's scale factor configurable

2023-04-03 12:31:53 -07:00

9 行

263 B

Swift

原始文件 永久链接 Blame 文件历史

 import CoreML
 @available(iOS 16.2, macOS 13.1, *)
 public protocol RandomSource {
     mutating func nextNormal(mean: Double, stdev: Double) -> Double
     mutating func normalShapedArray(_ shape: [Int], mean: Double, stdev: Double) -> MLShapedArray<Double>
 }