Skip to content

Commit 4655712

Browse files
finish tests (#909)
1 parent b35d88c commit 4655712

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_pipelines.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2013,7 +2013,7 @@ def test_stable_diffusion_inpaint_pipeline(self):
20132013
)
20142014
expected_image = np.array(expected_image, dtype=np.float32) / 255.0
20152015

2016-
model_id = "fusing/sd-inpaint-temp"
2016+
model_id = "runwayml/stable-diffusion-inpainting"
20172017
pipe = StableDiffusionInpaintPipeline.from_pretrained(
20182018
model_id,
20192019
safety_checker=self.dummy_safety_checker,
@@ -2050,11 +2050,11 @@ def test_stable_diffusion_inpaint_pipeline_fp16(self):
20502050
)
20512051
expected_image = load_image(
20522052
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
2053-
"/in_paint/yellow_cat_sitting_on_a_park_bench.png"
2053+
"/in_paint/yellow_cat_sitting_on_a_park_bench_fp16.png"
20542054
)
20552055
expected_image = np.array(expected_image, dtype=np.float32) / 255.0
20562056

2057-
model_id = "fusing/sd-inpaint-temp"
2057+
model_id = "runwayml/stable-diffusion-inpainting"
20582058
pipe = StableDiffusionInpaintPipeline.from_pretrained(
20592059
model_id,
20602060
revision="fp16",
@@ -2141,7 +2141,7 @@ def test_stable_diffusion_inpaint_pipeline_pndm(self):
21412141
expected_image = np.array(expected_image, dtype=np.float32) / 255.0
21422142

21432143
pndm = PNDMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True)
2144-
model_id = "fusing/sd-inpaint-temp"
2144+
model_id = "runwayml/stable-diffusion-inpainting"
21452145
pipe = StableDiffusionInpaintPipeline.from_pretrained(
21462146
model_id, safety_checker=self.dummy_safety_checker, scheduler=pndm
21472147
)

0 commit comments

Comments
 (0)