make it work with Instruments

Author: mortenjust

Example/maple-diffusion/ContentView.swift

@@ -132,6 +132,8 @@ struct ContentView: View {
                 case .ready:
                     print("model ready")
                     running = false
+                case .notStarted:
+                    break
                 }
             }
     }

Sources/MapleDiffusion/Functions.swift

@@ -555,10 +555,20 @@ func makeAuxUpsampler(graph: MPSGraph, xIn: MPSGraphTensor) -> MPSGraphTensor {
 }
 
 
-
 func tensorToCGImage(data: MPSGraphTensorData) -> CGImage {
     let shape = data.shape.map{$0.intValue}
     var imageArrayCPUBytes = [UInt8](repeating: 0, count: shape.reduce(1, *))
     data.mpsndarray().readBytes(&imageArrayCPUBytes, strideBytes: nil)
-    return CGImage(width: shape[2], height: shape[1], bitsPerComponent: 8, bitsPerPixel: 32, bytesPerRow: shape[2]*shape[3], space: CGColorSpaceCreateDeviceRGB(), bitmapInfo:  CGBitmapInfo(rawValue: CGBitmapInfo.byteOrder32Big.rawValue | CGImageAlphaInfo.noneSkipLast.rawValue), provider: CGDataProvider(data: NSData(bytes: &imageArrayCPUBytes, length: imageArrayCPUBytes.count))!, decode: nil, shouldInterpolate: true, intent: CGColorRenderingIntent.defaultIntent)!
+    
+    return CGImage(
+        width: shape[2], height: shape[1],
+        bitsPerComponent: 8, bitsPerPixel: 32,
+        bytesPerRow: shape[2]*shape[3],
+        space: CGColorSpaceCreateDeviceRGB(),
+        bitmapInfo:  CGBitmapInfo(rawValue: CGBitmapInfo.byteOrder32Big.rawValue | CGImageAlphaInfo.noneSkipLast.rawValue),
+        provider: CGDataProvider(data: NSData(bytes: &imageArrayCPUBytes,
+                                              length: imageArrayCPUBytes.count))!,
+        decode: nil,
+        shouldInterpolate: true,
+        intent: CGColorRenderingIntent.defaultIntent)!
 }

Sources/MapleDiffusion/MapleDiffusion.swift

@@ -358,6 +358,7 @@ public class MapleDiffusion : ObservableObject {
 
         // 3. Noise generation
         var latent = randomLatent(seed: seed)
+        
         let timesteps = Array<Int>(stride(from: 1, to: 1000, by: Int(1000 / steps)))
         completion(nil, 0.75 * 1 / Float(steps), "Starting diffusion...")
 
@@ -371,7 +372,8 @@ public class MapleDiffusion : ObservableObject {
             let tMPSData = MPSGraphTensorData(device: graphDevice!, data: tData, shape: [1], dataType: MPSDataType.int32)
             let tPrevData = [Int32(tsPrev)].withUnsafeBufferPointer {Data(buffer: $0)}
             let tPrevMPSData = MPSGraphTensorData(device: graphDevice!, data: tPrevData, shape: [1], dataType: MPSDataType.int32)
-            let guidanceScaleData = [Float16(guidanceScale)].withUnsafeBufferPointer {Data(buffer: $0)}
+            let guidanceScaleData = [Float32(guidanceScale)].withUnsafeBufferPointer {Data(buffer: $0)}
+
             let guidanceScaleMPSData = MPSGraphTensorData(device: graphDevice!, data: guidanceScaleData, shape: [1], dataType: MPSDataType.float16)
             let temb = tembGraph!.run(with: commandQueue!, feeds: [tembTIn!: tMPSData], targetTensors: [tembOut!], targetOperations: nil)[tembOut!]!
             let etaUncond: MPSGraphTensorData
@@ -473,3 +475,6 @@ public class MapleDiffusion : ObservableObject {
         }
     }
 }
+
+
+