update

a-r-r-o-w · a-r-r-o-w · commit 5bf2a5d15897 · 2025-01-13T14:42:59.000+01:00
diff --git a/tests/models/transformers/test_models_transformer_cogvideox.py b/tests/models/transformers/test_models_transformer_cogvideox.py
@@ -33,7 +33,7 @@ class CogVideoXTransformerTests(ModelTesterMixin, unittest.TestCase):
     model_class = CogVideoXTransformer3DModel
     main_input_name = "hidden_states"
     uses_custom_attn_processor = True
-    model_split_percents = [0.7, 0.6, 0.6]
+    model_split_percents = [0.7, 0.7, 0.8]
 
     @property
     def dummy_input(self):
