Skip to content

Commit 279f744

Browse files
controlnet integration tests num_inference_steps=3 (#2672)
1 parent ee71d9d commit 279f744

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/pipelines/stable_diffusion/test_stable_diffusion_controlnet.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ def test_canny(self):
322322
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bird_canny.png"
323323
)
324324

325-
output = pipe(prompt, image, generator=generator, output_type="np")
325+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
326326

327327
image = output.images[0]
328328

@@ -349,7 +349,7 @@ def test_depth(self):
349349
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/stormtrooper_depth.png"
350350
)
351351

352-
output = pipe(prompt, image, generator=generator, output_type="np")
352+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
353353

354354
image = output.images[0]
355355

@@ -376,7 +376,7 @@ def test_hed(self):
376376
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/man_hed.png"
377377
)
378378

379-
output = pipe(prompt, image, generator=generator, output_type="np")
379+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
380380

381381
image = output.images[0]
382382

@@ -403,7 +403,7 @@ def test_mlsd(self):
403403
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/room_mlsd.png"
404404
)
405405

406-
output = pipe(prompt, image, generator=generator, output_type="np")
406+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
407407

408408
image = output.images[0]
409409

@@ -430,7 +430,7 @@ def test_normal(self):
430430
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/cute_toy_normal.png"
431431
)
432432

433-
output = pipe(prompt, image, generator=generator, output_type="np")
433+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
434434

435435
image = output.images[0]
436436

@@ -457,7 +457,7 @@ def test_openpose(self):
457457
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/pose.png"
458458
)
459459

460-
output = pipe(prompt, image, generator=generator, output_type="np")
460+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
461461

462462
image = output.images[0]
463463

@@ -484,7 +484,7 @@ def test_scribble(self):
484484
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bag_scribble.png"
485485
)
486486

487-
output = pipe(prompt, image, generator=generator, output_type="np")
487+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
488488

489489
image = output.images[0]
490490

@@ -511,7 +511,7 @@ def test_seg(self):
511511
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/house_seg.png"
512512
)
513513

514-
output = pipe(prompt, image, generator=generator, output_type="np")
514+
output = pipe(prompt, image, generator=generator, output_type="np", num_inference_steps=3)
515515

516516
image = output.images[0]
517517

@@ -581,7 +581,7 @@ def test_pose_and_canny(self):
581581
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/pose.png"
582582
)
583583

584-
output = pipe(prompt, [image_pose, image_canny], generator=generator, output_type="np")
584+
output = pipe(prompt, [image_pose, image_canny], generator=generator, output_type="np", num_inference_steps=3)
585585

586586
image = output.images[0]
587587

0 commit comments

Comments
 (0)