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

[Marigold tests] add is_flaky decorator to some Marigold tests (#8696)

Browse Source 
okay
This commit is contained in:
Sayak Paul
2024-06-25 21:57:28 +05:30
committed by GitHub
parent c6e08ecd46
commit f088027e93
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/pipelines/marigold/test_marigold_depth.py
View File

@@ -34,6 +34,7 @@ from diffusers import (
from diffusers.utils.testing_utils import (
enable_full_determinism,
floats_tensor,
is_flaky,
load_image,
require_torch_gpu,
slow,
@@ -232,6 +233,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
match_input_resolution=True,
)
@is_flaky
def test_marigold_depth_dummy_G0_S1_P32_E3_B1_M1(self):
self._test_marigold_depth(
generator_seed=0,
@@ -244,6 +246,7 @@ class MarigoldDepthPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
match_input_resolution=True,
)
@is_flaky
def test_marigold_depth_dummy_G0_S1_P32_E4_B2_M1(self):
self._test_marigold_depth(
generator_seed=0,