Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

finish tests (#909)

This commit is contained in:
Patrick von Platen 2022-10-19 16:36:51 +02:00 committed by GitHub
parent b35d88c536
commit 46557121e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_pipelines.py
View File

@ -2013,7 +2013,7 @@ class PipelineTesterMixin(unittest.TestCase):
) )
expected_image = np.array(expected_image, dtype=np.float32) / 255.0 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( pipe = StableDiffusionInpaintPipeline.from_pretrained(
model_id, model_id,
safety_checker=self.dummy_safety_checker, safety_checker=self.dummy_safety_checker,
@ -2050,11 +2050,11 @@ class PipelineTesterMixin(unittest.TestCase):
) )
expected_image = load_image( expected_image = load_image(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main" "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 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( pipe = StableDiffusionInpaintPipeline.from_pretrained(
model_id, model_id,
revision="fp16", revision="fp16",
@ -2141,7 +2141,7 @@ class PipelineTesterMixin(unittest.TestCase):
expected_image = np.array(expected_image, dtype=np.float32) / 255.0 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) 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( pipe = StableDiffusionInpaintPipeline.from_pretrained(
model_id, safety_checker=self.dummy_safety_checker, scheduler=pndm model_id, safety_checker=self.dummy_safety_checker, scheduler=pndm
) )