Skip to content

Commit d89e785

Browse files
committed
Update test_pag_controlnet_sd_img2img.py
1 parent ce6c10f commit d89e785

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/pag/test_pag_controlnet_sd_img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def test_pag_disable_enable(self):
177177
components = self.get_dummy_components()
178178

179179
# base pipeline (expect same output when pag is disabled)
180-
pipe_sd = StableDiffusionControlNetPipeline(**components)
180+
pipe_sd = StableDiffusionControlNetImg2ImgPipeline(**components)
181181
pipe_sd = pipe_sd.to(device)
182182
pipe_sd.set_progress_bar_config(disable=None)
183183

0 commit comments

Comments
 (0)