From 5bf9c71081003d72c96b9a48edb37ffaa42782cd Mon Sep 17 00:00:00 2001
From: Timothy Kautz <tim@timothykautz.com>
Date: Thu, 9 Feb 2023 00:21:19 -0800
Subject: [PATCH] Add fix for Jpegs

---
 swift/StableDiffusionCLI/main.swift | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/swift/StableDiffusionCLI/main.swift b/swift/StableDiffusionCLI/main.swift
index 6b46046..c5ae31a 100644
--- a/swift/StableDiffusionCLI/main.swift
+++ b/swift/StableDiffusionCLI/main.swift
@@ -7,6 +7,7 @@ import CoreML
 import Foundation
 import StableDiffusion
 import UniformTypeIdentifiers
+import Cocoa
 
 @available(iOS 16.2, macOS 13.1, *)
 struct StableDiffusionSample: ParsableCommand {
@@ -97,11 +98,8 @@ struct StableDiffusionSample: ParsableCommand {
             do {
                 let imageData = try Data(contentsOf: imageURL)
                 guard
-                    let imgDataProvider = CGDataProvider(data: imageData as CFData),
-                    let loadedImage = CGImage(
-                        pngDataProviderSource: imgDataProvider,
-                        decode: nil, shouldInterpolate: false,
-                        intent: CGColorRenderingIntent.defaultIntent)
+                    let nsImage = NSImage(data: imageData),
+                    let loadedImage = nsImage.cgImage(forProposedRect: nil, context: nil, hints: nil)
                 else {
                     throw RunError.resources("Starting Image not available \(resourcePath)")
                 }