Skip to content

GitLab

Unverified Commit 4c26cb9c authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Tests] Fix slow tests (#2846)

parent 1d7b4b60
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/pipelines/stable_unclip/test_stable_unclip_img2img.py
View file @ 4c26cb9c
...@@ -197,7 +197,7 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase): ...@@ -197,7 +197,7 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase):
pipe.enable_sequential_cpu_offload() pipe.enable_sequential_cpu_offload()
generator = torch.Generator(device="cpu").manual_seed(0) generator = torch.Generator(device="cpu").manual_seed(0)
output = pipe("anime turle", image=input_image, generator=generator, output_type="np") output = pipe(input_image, "anime turle", generator=generator, output_type="np")
image = output.images[0] image = output.images[0]
...@@ -225,7 +225,7 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase): ...@@ -225,7 +225,7 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase):
pipe.enable_sequential_cpu_offload() pipe.enable_sequential_cpu_offload()
generator = torch.Generator(device="cpu").manual_seed(0) generator = torch.Generator(device="cpu").manual_seed(0)
output = pipe("anime turle", image=input_image, generator=generator, output_type="np") output = pipe(input_image, "anime turle", generator=generator, output_type="np")
image = output.images[0] image = output.images[0]
...@@ -251,8 +251,8 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase): ...@@ -251,8 +251,8 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase):
pipe.enable_sequential_cpu_offload() pipe.enable_sequential_cpu_offload()
_ = pipe( _ = pipe(
input_image,
"anime turtle", "anime turtle",
image=input_image,
num_inference_steps=2, num_inference_steps=2,
output_type="np", output_type="np",
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment