Skip to content

Commit 4ca68f2

Browse files
committed
skipping ZImage DiT tests
1 parent d910a26 commit 4ca68f2

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/models/transformers/test_models_transformer_z_image.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121

2222
from diffusers import ZImageTransformer2DModel
2323

24-
from ...testing_utils import torch_device
24+
from ...testing_utils import IS_GITHUB_ACTIONS, torch_device
2525
from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin
2626

2727

@@ -36,6 +36,10 @@
3636
torch.backends.cuda.matmul.allow_tf32 = False
3737

3838

39+
@unittest.skipIf(
40+
IS_GITHUB_ACTIONS,
41+
reason="Skipping test-suite inside the CI because the model has `torch.empty()` inside of it during init and we don't have a clear way to override it in the modeling tests.",
42+
)
3943
class ZImageTransformerTests(ModelTesterMixin, unittest.TestCase):
4044
model_class = ZImageTransformer2DModel
4145
main_input_name = "x"

0 commit comments

Comments
 (0)