From 4f01e1b42a7a2429bd3c24ab19c2319638e8c3d0 Mon Sep 17 00:00:00 2001
From: JustinMeans <46542161+JustinMeans@users.noreply.github.com>
Date: Tue, 20 Dec 2022 18:04:21 -0700
Subject: [PATCH] Adds negative prompt argument to CLI

---
 swift/StableDiffusionCLI/main.swift | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/swift/StableDiffusionCLI/main.swift b/swift/StableDiffusionCLI/main.swift
index df0ad92..5cbe627 100644
--- a/swift/StableDiffusionCLI/main.swift
+++ b/swift/StableDiffusionCLI/main.swift
@@ -19,6 +19,9 @@ struct StableDiffusionSample: ParsableCommand {
     @Argument(help: "Input string prompt")
     var prompt: String
 
+    @Option(help: "Input string negative prompt")
+    var negativePrompt: String
+
     @Option(
         help: ArgumentHelp(
             "Path to stable diffusion resources.",
@@ -85,6 +88,7 @@ struct StableDiffusionSample: ParsableCommand {
 
         let images = try pipeline.generateImages(
             prompt: prompt,
+            negativePrompt: negativePrompt,
             imageCount: imageCount,
             stepCount: stepCount,
             seed: seed,