Supports SDXL Lightning

Author: digitalbrain79

Diffusion/Common/Pipeline/Pipeline.swift

@@ -76,9 +76,11 @@ class Pipeline {
         negativePrompt: String = "",
         scheduler: StableDiffusionScheduler,
         numInferenceSteps stepCount: Int = 50,
+        //numInferenceSteps stepCount: Int = 4,
         seed: UInt32 = 0,
         numPreviews previewCount: Int = 5,
         guidanceScale: Float = 7.5,
+        //guidanceScale: Float = 0,
         disableSafety: Bool = false
     ) throws -> GenerationResult {
         let beginDate = Date()

Diffusion/Common/Pipeline/PipelineLoader.swift

@@ -93,14 +93,16 @@ extension PipelineLoader {
 
     var packagesFilename: String { (filename as NSString).deletingPathExtension }
 
-    var compiledURL: URL { downloadedURL.deletingLastPathComponent().appendingPathComponent(packagesFilename)  }
+    //var compiledURL: URL { downloadedURL.deletingLastPathComponent().appendingPathComponent(packagesFilename)  }
+    var compiledURL: URL { URL(fileURLWithPath: Bundle.main.resourcePath!)  }
 
     var downloaded: Bool {
         return FileManager.default.fileExists(atPath: downloadedURL.path)
     }
 
     var ready: Bool {
-        return FileManager.default.fileExists(atPath: compiledURL.path)
+        //return FileManager.default.fileExists(atPath: compiledURL.path)
+        true
     }
 
     var variant: AttentionVariant {
@@ -165,6 +167,11 @@ extension PipelineLoader {
         let beginDate = Date()
         let configuration = MLModelConfiguration()
         configuration.computeUnits = computeUnits
+        
+        print(configuration.computeUnits.rawValue)
+        print(model.reduceMemory)
+        print(compiledURL)
+        
         let pipeline: StableDiffusionPipelineProtocol
         if model.isXL {
             if #available(macOS 14.0, iOS 17.0, *) {

Diffusion/Common/State.swift

@@ -12,7 +12,7 @@ import StableDiffusion
 import CoreML
 
 let DEFAULT_MODEL = ModelInfo.sd3
-let DEFAULT_PROMPT = "Labrador in the style of Vermeer"
+let DEFAULT_PROMPT = "a photo of puppy"
 
 enum GenerationState {
     case startup
@@ -130,8 +130,10 @@ class Settings {
             Keys.computeUnits.rawValue: -1,      // Use default
             Keys.prompt.rawValue: DEFAULT_PROMPT,
             Keys.negativePrompt.rawValue: "",
-            Keys.guidanceScale.rawValue: 7.5,
-            Keys.stepCount.rawValue: 25,
+            //Keys.guidanceScale.rawValue: 7.5,
+            Keys.guidanceScale.rawValue: 0,
+            //Keys.stepCount.rawValue: 25,
+            Keys.stepCount.rawValue: 4,
             Keys.previewCount.rawValue: 5,
             Keys.seed.rawValue: 0
         ])

Diffusion/Info.plist

@@ -1,10 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
 <plist version="1.0">
-	<dict>
-	    <key>NSPhotoLibraryAddUsageDescription</key>
-    	<string>To be able to save generated images to your Photo Library, you’ll need to allow this.</string>
-	    <key>CFBundleDisplayName</key>
-    	<string>Diffusers</string>
-	</dict>
+<dict/>
 </plist>