diff --git a/tests/pipelines/amused/test_amused_img2img.py b/tests/pipelines/amused/test_amused_img2img.py index c03826e6abc9..a7b4b0141473 100644 --- a/tests/pipelines/amused/test_amused_img2img.py +++ b/tests/pipelines/amused/test_amused_img2img.py @@ -162,9 +162,7 @@ def test_amused_256(self): assert np.abs(image_slice - expected_slice).max() < 1e-2 def test_amused_256_fp16(self): - pipe = AmusedImg2ImgPipeline.from_pretrained( - "amused/amused-256", torch_dtype=torch.float16, variant="fp16" - ) + pipe = AmusedImg2ImgPipeline.from_pretrained("amused/amused-256", torch_dtype=torch.float16, variant="fp16") pipe.to(torch_device) image = ( @@ -213,9 +211,7 @@ def test_amused_512(self): assert np.abs(image_slice - expected_slice).max() < 0.1 def test_amused_512_fp16(self): - pipe = AmusedImg2ImgPipeline.from_pretrained( - "amused/amused-512", variant="fp16", torch_dtype=torch.float16 - ) + pipe = AmusedImg2ImgPipeline.from_pretrained("amused/amused-512", variant="fp16", torch_dtype=torch.float16) pipe.to(torch_device) image = ( diff --git a/tests/pipelines/amused/test_amused_inpaint.py b/tests/pipelines/amused/test_amused_inpaint.py index 93e43e576bf8..658736b12f68 100644 --- a/tests/pipelines/amused/test_amused_inpaint.py +++ b/tests/pipelines/amused/test_amused_inpaint.py @@ -174,9 +174,7 @@ def test_amused_256(self): assert np.abs(image_slice - expected_slice).max() < 0.1 def test_amused_256_fp16(self): - pipe = AmusedInpaintPipeline.from_pretrained( - "amused/amused-256", variant="fp16", torch_dtype=torch.float16 - ) + pipe = AmusedInpaintPipeline.from_pretrained("amused/amused-256", variant="fp16", torch_dtype=torch.float16) pipe.to(torch_device) image = ( @@ -242,9 +240,7 @@ def test_amused_512(self): assert np.abs(image_slice - expected_slice).max() < 0.05 def test_amused_512_fp16(self): - pipe = AmusedInpaintPipeline.from_pretrained( - "amused/amused-512", variant="fp16", torch_dtype=torch.float16 - ) + pipe = AmusedInpaintPipeline.from_pretrained("amused/amused-512", variant="fp16", torch_dtype=torch.float16) pipe.to(torch_device) image = (