From e100b8f2a3f7d7d88ac0ca6c33a47a0dd215d8f1 Mon Sep 17 00:00:00 2001 From: Daniel Gu Date: Sat, 13 Dec 2025 10:34:11 +0100 Subject: [PATCH] Rename LTX 2 compile test class to have LTX2 --- tests/models/transformers/test_models_transformer_ltx2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/transformers/test_models_transformer_ltx2.py b/tests/models/transformers/test_models_transformer_ltx2.py index d67789acca..fc089e6190 100644 --- a/tests/models/transformers/test_models_transformer_ltx2.py +++ b/tests/models/transformers/test_models_transformer_ltx2.py @@ -108,7 +108,7 @@ class LTX2TransformerTests(ModelTesterMixin, unittest.TestCase): super().test_gradient_checkpointing_is_applied(expected_set=expected_set) -class LTXTransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase): +class LTX2TransformerCompileTests(TorchCompileTesterMixin, unittest.TestCase): model_class = LTX2VideoTransformer3DModel def prepare_init_args_and_inputs_for_common(self):