Skip to content

Commit 98cc6d0

Browse files
[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 <[email protected]>
1 parent c372615 commit 98cc6d0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/transformers/test_models_transformer_ltx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,13 @@
2020
from diffusers import LTXVideoTransformer3DModel
2121
from diffusers.utils.testing_utils import enable_full_determinism, torch_device
2222

23-
from ..test_modeling_common import ModelTesterMixin
23+
from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin
2424

2525

2626
enable_full_determinism()
2727

2828

29-
class LTXTransformerTests(ModelTesterMixin, unittest.TestCase):
29+
class LTXTransformerTests(ModelTesterMixin, TorchCompileTesterMixin, unittest.TestCase):
3030
model_class = LTXVideoTransformer3DModel
3131
main_input_name = "hidden_states"
3232
uses_custom_attn_processor = True

0 commit comments

Comments
 (0)