Fix img2img starting image shape (#168)

vzsg · web-flow · commit fbef6c38264f · 2023-04-26T13:01:45.000-07:00
diff --git a/swift/StableDiffusion/pipeline/CGImage+vImage.swift b/swift/StableDiffusion/pipeline/CGImage+vImage.swift
@@ -111,7 +111,7 @@ extension CGImage {
             
             let imageData = redData + greenData + blueData
 
-            let shapedArray = MLShapedArray<Float32>(data: imageData, shape: [1, 3, self.width, self.height])
+            let shapedArray = MLShapedArray<Float32>(data: imageData, shape: [1, 3, self.height, self.width])
             
             return shapedArray
     }

Original file line number	Diff line number	Diff line change
`@@ -111,7 +111,7 @@ extension CGImage {`
`111`	`111`
`112`	`112`	`let imageData = redData + greenData + blueData`
`113`	`113`
`114`		`- let shapedArray = MLShapedArray<Float32>(data: imageData, shape: [1, 3, self.width, self.height])`
	`114`	`+ let shapedArray = MLShapedArray<Float32>(data: imageData, shape: [1, 3, self.height, self.width])`
`115`	`115`
`116`	`116`	`return shapedArray`
`117`	`117`	`}`