lint

huggingface · Dec 26, 2023 · e154eb6 · e154eb6
1 parent b21454b
commit e154eb6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git 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
@@ -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 = (