@@ -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