Skip to content

Commit

Permalink
Fixes apple#160
Browse files Browse the repository at this point in the history
  • Loading branch information
atiorh committed Apr 24, 2023
1 parent ce0be82 commit 412c679
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions python_coreml_stable_diffusion/pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -580,6 +580,7 @@ def main(args):
num_inference_steps=args.num_inference_steps,
guidance_scale=args.guidance_scale,
controlnet_cond=controlnet_cond,
negative_prompt=args.negative_prompt,
)

out_path = get_image_path(args)
Expand Down Expand Up @@ -646,6 +647,10 @@ def main(args):
type=str,
help=("Image paths for ControlNet inputs. "
"Please enter images corresponding to each controlnet provided at --controlnet option in same order."))
parser.add_argument(
"--negative-prompt",
default=None,
help="The negative text prompt to be used for text-to-image generation.")

args = parser.parse_args()
main(args)

0 comments on commit 412c679

Please sign in to comment.