Adds negative prompt argument to CLI

pull/61/head
JustinMeans 3 years ago committed by GitHub
parent e2924a8a8d
commit 4f01e1b42a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -19,6 +19,9 @@ struct StableDiffusionSample: ParsableCommand {
@Argument(help: "Input string prompt") @Argument(help: "Input string prompt")
var prompt: String var prompt: String
@Option(help: "Input string negative prompt")
var negativePrompt: String
@Option( @Option(
help: ArgumentHelp( help: ArgumentHelp(
"Path to stable diffusion resources.", "Path to stable diffusion resources.",
@ -85,6 +88,7 @@ struct StableDiffusionSample: ParsableCommand {
let images = try pipeline.generateImages( let images = try pipeline.generateImages(
prompt: prompt, prompt: prompt,
negativePrompt: negativePrompt,
imageCount: imageCount, imageCount: imageCount,
stepCount: stepCount, stepCount: stepCount,
seed: seed, seed: seed,

Loading…
Cancel
Save