Adds negative prompt argument to CLI

1 year ago · 4f01e1b42a
parent e2924a8a8d
commit 4f01e1b42a
1 changed files with 4 additions and 0 deletions
--- 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,