From 98cc6d05e478533b09620192ab44fb752f563c2f Mon Sep 17 00:00:00 2001 From: Evan Han Date: Mon, 12 May 2025 22:56:15 +0900 Subject: [PATCH] [test_models_transformer_ltx.py] help us test torch.compile() for impactful models (#11512) * Update test_models_transformer_ltx.py * Update test_models_transformer_ltx.py * Update test_models_transformer_ltx.py * Update test_models_transformer_ltx.py --------- Co-authored-by: Sayak Paul --- tests/models/transformers/test_models_transformer_ltx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/transformers/test_models_transformer_ltx.py b/tests/models/transformers/test_models_transformer_ltx.py index 128bf04155..8649ce97a5 100644 --- a/tests/models/transformers/test_models_transformer_ltx.py +++ b/tests/models/transformers/test_models_transformer_ltx.py @@ -20,13 +20,13 @@ import torch from diffusers import LTXVideoTransformer3DModel from diffusers.utils.testing_utils import enable_full_determinism, torch_device -from ..test_modeling_common import ModelTesterMixin +from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin enable_full_determinism() -class LTXTransformerTests(ModelTesterMixin, unittest.TestCase): +class LTXTransformerTests(ModelTesterMixin, TorchCompileTesterMixin, unittest.TestCase): model_class = LTXVideoTransformer3DModel main_input_name = "hidden_states" uses_custom_attn_processor = True