AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Tests] Fix slow tests (#2846)

Browse Source
This commit is contained in:
Patrick von Platen
2023-03-27 19:45:49 +02:00
committed by GitHub
parent 1d7b4b60b7
commit 4c26cb9cc8
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/pipelines/stable_unclip/test_stable_unclip_img2img.py
View File

@@ -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",
)