From 46557121e6421e4d4e7f294584eac4961ef7c728 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 19 Oct 2022 16:36:51 +0200 Subject: [PATCH] finish tests (#909) --- tests/test_pipelines.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index e40d01e1..2a581e61 100644 --- a/tests/test_pipelines.py +++ b/tests/test_pipelines.py @@ -2013,7 +2013,7 @@ class PipelineTesterMixin(unittest.TestCase): ) expected_image = np.array(expected_image, dtype=np.float32) / 255.0 - model_id = "fusing/sd-inpaint-temp" + model_id = "runwayml/stable-diffusion-inpainting" pipe = StableDiffusionInpaintPipeline.from_pretrained( model_id, safety_checker=self.dummy_safety_checker, @@ -2050,11 +2050,11 @@ class PipelineTesterMixin(unittest.TestCase): ) expected_image = load_image( "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main" - "/in_paint/yellow_cat_sitting_on_a_park_bench.png" + "/in_paint/yellow_cat_sitting_on_a_park_bench_fp16.png" ) expected_image = np.array(expected_image, dtype=np.float32) / 255.0 - model_id = "fusing/sd-inpaint-temp" + model_id = "runwayml/stable-diffusion-inpainting" pipe = StableDiffusionInpaintPipeline.from_pretrained( model_id, revision="fp16", @@ -2141,7 +2141,7 @@ class PipelineTesterMixin(unittest.TestCase): expected_image = np.array(expected_image, dtype=np.float32) / 255.0 pndm = PNDMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", skip_prk_steps=True) - model_id = "fusing/sd-inpaint-temp" + model_id = "runwayml/stable-diffusion-inpainting" pipe = StableDiffusionInpaintPipeline.from_pretrained( model_id, safety_checker=self.dummy_safety_checker, scheduler=pndm )