diff --git a/python_coreml_stable_diffusion/pipeline.py b/python_coreml_stable_diffusion/pipeline.py index 61f9bbf..442d469 100644 --- a/python_coreml_stable_diffusion/pipeline.py +++ b/python_coreml_stable_diffusion/pipeline.py @@ -480,6 +480,7 @@ def main(args): height=coreml_pipe.height, width=coreml_pipe.width, num_inference_steps=args.num_inference_steps, + guidance_scale=args.guidance_scale ) out_path = get_image_path(args) @@ -529,6 +530,11 @@ if __name__ == "__main__": default=50, type=int, help="The number of iterations the unet model will be executed throughout the reverse diffusion process") + parser.add_argument( + "--guidance-scale", + default=7.5, + type=float, + help="Controls the influence of the text prompt on sampling process (0=random images)") args = parser.parse_args() main(args)