def run_inference()

in run_benchmark_pixart.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

def run_inference(pipe, args):
    _ = pipe(
        prompt=args.prompt,
        num_inference_steps=args.num_inference_steps,
        num_images_per_prompt=args.batch_size,
    )