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):
pipe.enable_sequential_cpu_offload()
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]
......@@ -225,7 +225,7 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase):
pipe.enable_sequential_cpu_offload()
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]
......@@ -251,8 +251,8 @@ class StableUnCLIPImg2ImgPipelineIntegrationTests(unittest.TestCase):
pipe.enable_sequential_cpu_offload()
_ = pipe(
input_image,
"anime turtle",
image=input_image,
num_inference_steps=2,
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