LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2025-12-06 12:34:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

up

Browse Source
This commit is contained in:
sayakpaul
2025-11-29 10:42:38 +05:30
parent a11cdd260b
commit bca3e27c96
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/models/transformers/test_models_transformer_z_image.py
View File

@@ -21,7 +21,7 @@ import torch
from diffusers import ZImageTransformer2DModel
from ...testing_utils import torch_device
from ...testing_utils import is_flaky, torch_device
from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin
@@ -139,6 +139,22 @@ class ZImageTransformerTests(ModelTesterMixin, unittest.TestCase):
def test_group_offloading_with_disk(self):
super().test_group_offloading_with_disk()
@is_flaky(max_attempts=10)
def test_auto_model(self):
super().test_auto_model()
@is_flaky(max_attempts=10)
def test_determinism(self):
super().test_determinism()
@is_flaky(max_attempts=10)
def test_from_save_pretrained(self):
super().test_from_save_pretrained()
@is_flaky(max_attempts=10)
def test_from_save_pretrained_variant(self):
super().test_from_save_pretrained_variant()
class Flux2TransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase):
model_class = ZImageTransformer2DModel