Skip to content

Commit 46fb70d

Browse files
committed
Update test_models_transformer_ltx.py
1 parent b21a753 commit 46fb70d

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
@@ -27,13 +27,13 @@
2727
torch_device,
2828
)
2929

30-
from ..test_modeling_common import ModelTesterMixin
30+
from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin
3131

3232

3333
enable_full_determinism()
3434

3535

36-
class LTXTransformerTests(ModelTesterMixin, unittest.TestCase):
36+
class LTXTransformerTests(ModelTesterMixin, TorchCompileTesterMixin, unittest.TestCase):
3737
model_class = LTXVideoTransformer3DModel
3838
main_input_name = "hidden_states"
3939
uses_custom_attn_processor = True

0 commit comments

Comments
 (0)