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

[CI] Fix SANA tests (#11756)

Browse Source 
update
This commit is contained in:
Dhruv Nair
2025-06-19 13:36:02 +02:00
committed by GitHub
parent fc51583c8a
commit 0c11c8c1ac
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/pipelines/sana/test_sana_controlnet.py
View File

@@ -30,6 +30,7 @@ from diffusers.utils.testing_utils import (
enable_full_determinism,
torch_device,
)
from diffusers.utils.torch_utils import randn_tensor
from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_IMAGE_PARAMS, TEXT_TO_IMAGE_PARAMS
from ..test_pipelines_common import PipelineTesterMixin, to_np
@@ -151,7 +152,7 @@ class SanaControlNetPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
else:
generator = torch.Generator(device=device).manual_seed(seed)
control_image = torch.randn(1, 3, 32, 32, generator=generator)
control_image = randn_tensor((1, 3, 32, 32), generator=generator, device=device)
inputs = {
"prompt": "",
"negative_prompt": "",

3
tests/pipelines/sana/test_sana_sprint_img2img.py
View File

@@ -24,6 +24,7 @@ from diffusers.utils.testing_utils import (
enable_full_determinism,
torch_device,
)
from diffusers.utils.torch_utils import randn_tensor
from ..pipeline_params import (
IMAGE_TO_IMAGE_IMAGE_PARAMS,
@@ -137,7 +138,7 @@ class SanaSprintImg2ImgPipelineFastTests(PipelineTesterMixin, unittest.TestCase)
generator = torch.manual_seed(seed)
else:
generator = torch.Generator(device=device).manual_seed(seed)
image = torch.randn(1, 3, 32, 32, generator=generator)
image = randn_tensor((1, 3, 32, 32), generator=generator, device=device)
inputs = {
"prompt": "",
"image": image,