Merge pull request #1 from chigozienri/negative-prompt

Add negative_prompt
lucataco · Nov 10, 2023 · d4ea9e1 · d4ea9e1
2 parents 78b1a47 + c186628
commit d4ea9e1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -41,6 +41,10 @@ def predict(
             description="Prompt",
             default="photo of a beautiful girl wearing casual shirt in a garden"
         ),
+        negative_prompt: str = Input(
+            description="Negative Prompt",
+            default="monochrome, lowres, bad anatomy, worst quality, low quality"
+        ),
         num_outputs: int = Input(
             description="Number of images to output.",
             ge=1,
@@ -70,7 +74,8 @@ def predict(
             num_samples=num_outputs,
             num_inference_steps=num_inference_steps,
             seed=seed,
-            prompt=prompt
+            prompt=prompt,
+            negative_prompt=negative_prompt
         )
 
         output_paths = []