From a11cdd260b6a7c2829bd0a5eb19477c4ad0c3801 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sat, 29 Nov 2025 08:53:12 +0530 Subject: [PATCH] up --- tests/models/transformers/test_models_transformer_z_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/transformers/test_models_transformer_z_image.py b/tests/models/transformers/test_models_transformer_z_image.py index cae1173a72..4fcb7728df 100644 --- a/tests/models/transformers/test_models_transformer_z_image.py +++ b/tests/models/transformers/test_models_transformer_z_image.py @@ -40,7 +40,7 @@ class ZImageTransformerTests(ModelTesterMixin, unittest.TestCase): model_class = ZImageTransformer2DModel main_input_name = "x" # We override the items here because the transformer under consideration is small. - model_split_percents = [0.8, 0.8, 0.9] + model_split_percents = [0.9, 0.9, 0.9] def prepare_dummy_input(self, height=16, width=16): batch_size = 1