From 24e05b441fbd1e002dc5e9868ce15f426cbf0fc4 Mon Sep 17 00:00:00 2001
From: "yexulei.k" <yexulei.k@bytedance.com>
Date: Sat, 17 Dec 2022 14:09:40 +0800
Subject: [PATCH] Update CLI to use @main

---
 .../{main.swift => StableDiffusionSample.swift}            | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)
 rename swift/StableDiffusionCLI/{main.swift => StableDiffusionSample.swift} (98%)

diff --git a/swift/StableDiffusionCLI/main.swift b/swift/StableDiffusionCLI/StableDiffusionSample.swift
similarity index 98%
rename from swift/StableDiffusionCLI/main.swift
rename to swift/StableDiffusionCLI/StableDiffusionSample.swift
index 4da8e62..decea05 100644
--- a/swift/StableDiffusionCLI/main.swift
+++ b/swift/StableDiffusionCLI/StableDiffusionSample.swift
@@ -9,6 +9,7 @@ import StableDiffusion
 import UniformTypeIdentifiers
 
 @available(iOS 16.2, macOS 13.1, *)
+@main
 struct StableDiffusionSample: ParsableCommand {
 
     static let configuration = CommandConfiguration(
@@ -205,9 +206,3 @@ enum SchedulerOption: String, ExpressibleByArgument {
         }
     }
 }
-
-if #available(iOS 16.2, macOS 13.1, *) {
-    StableDiffusionSample.main()
-} else {
-    print("Unsupported OS")
-}