From a2f09ef0ab1dac3b457bbfecfde2190cd2342b15 Mon Sep 17 00:00:00 2001 From: Will Berman Date: Tue, 14 Mar 2023 20:03:36 +0000 Subject: [PATCH] controlnet integration tests num_inference_steps=3 --- .../test_stable_diffusion_controlnet.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py index c235b493340d..d556e6318f43 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py @@ -322,7 +322,7 @@ def test_canny(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bird_canny.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -349,7 +349,7 @@ def test_depth(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/stormtrooper_depth.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -376,7 +376,7 @@ def test_hed(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/man_hed.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -403,7 +403,7 @@ def test_mlsd(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/room_mlsd.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -430,7 +430,7 @@ def test_normal(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/cute_toy_normal.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -457,7 +457,7 @@ def test_openpose(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/pose.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -484,7 +484,7 @@ def test_scribble(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bag_scribble.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -511,7 +511,7 @@ def test_seg(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/house_seg.png" ) - output = pipe(prompt, image, generator=generator, output_type="np") + output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3) image = output.images[0] @@ -581,7 +581,7 @@ def test_pose_and_canny(self): "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/pose.png" ) - output = pipe(prompt, [image_pose, image_canny], generator=generator, output_type="np") + output = pipe(prompt, [image_pose, image_canny], generator=generator, output_type="np", num_inference_steps=3) image = output.images[0]