Update Decoder.swift (#249)

Ryu-ga · Ryu-ga · web-flow · commit c5063225aa14 · 2023-09-01T11:50:01.000-07:00
* Update Decoder.swift

* Update Encoder.swift

---------

Co-authored-by: Ryu-ga &lt;frech5679@icloud.com&gt;
diff --git a/swift/StableDiffusion/pipeline/Decoder.swift b/swift/StableDiffusion/pipeline/Decoder.swift
@@ -64,7 +64,7 @@ public struct Decoder: ResourceManaging {
             let result = results.features(at: i)
             let outputName = result.featureNames.first!
             let output = result.featureValue(for: outputName)!.multiArrayValue!
-            return try CGImage.fromShapedArray(MLShapedArray<Float32>(output))
+            return try CGImage.fromShapedArray(MLShapedArray<Float32>(converting: output))
         }
 
         return images
diff --git a/swift/StableDiffusion/pipeline/Encoder.swift b/swift/StableDiffusion/pipeline/Encoder.swift
@@ -63,7 +63,7 @@ public struct Encoder: ResourceManaging {
         }
         let outputName = result.featureNames.first!
         let outputValue = result.featureValue(for: outputName)!.multiArrayValue!
-        let output = MLShapedArray<Float32>(outputValue)
+        let output = MLShapedArray<Float32>(converting: outputValue)
         
         // DiagonalGaussianDistribution
         let mean = output[0][0..<4]

Original file line number	Diff line number	Diff line change
`@@ -64,7 +64,7 @@ public struct Decoder: ResourceManaging {`
`64`	`64`	`let result = results.features(at: i)`
`65`	`65`	`let outputName = result.featureNames.first!`
`66`	`66`	`let output = result.featureValue(for: outputName)!.multiArrayValue!`
`67`		`- return try CGImage.fromShapedArray(MLShapedArray<Float32>(output))`
	`67`	`+ return try CGImage.fromShapedArray(MLShapedArray<Float32>(converting: output))`
`68`	`68`	`}`
`69`	`69`
`70`	`70`	`return images`
Original file line number	Diff line number	Diff line change
`@@ -63,7 +63,7 @@ public struct Encoder: ResourceManaging {`
`63`	`63`	`}`
`64`	`64`	`let outputName = result.featureNames.first!`
`65`	`65`	`let outputValue = result.featureValue(for: outputName)!.multiArrayValue!`
`66`		`- let output = MLShapedArray<Float32>(outputValue)`
	`66`	`+ let output = MLShapedArray<Float32>(converting: outputValue)`
`67`	`67`
`68`	`68`	`// DiagonalGaussianDistribution`
`69`	`69`	`let mean = output[0][0..<4]`