diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index cbbb7cf9b..84744fb13 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -56,7 +56,7 @@ def run_txt2img_pipeline( rng = np.random.RandomState(params.seed) progress = job.get_progress_callback() - result = pipe.txt2img( + result = pipe.text2img( params.prompt, height=size.height, width=size.width,