Unverified Commit 98cc6d05 authored by Evan Han's avatar Evan Han Committed by GitHub
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: default avatarSayak Paul <spsayakpaul@gmail.com>
parent c3726153
...@@ -20,13 +20,13 @@ import torch ...@@ -20,13 +20,13 @@ import torch
from diffusers import LTXVideoTransformer3DModel from diffusers import LTXVideoTransformer3DModel
from diffusers.utils.testing_utils import enable_full_determinism, torch_device 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() enable_full_determinism()
class LTXTransformerTests(ModelTesterMixin, unittest.TestCase): class LTXTransformerTests(ModelTesterMixin, TorchCompileTesterMixin, unittest.TestCase):
model_class = LTXVideoTransformer3DModel model_class = LTXVideoTransformer3DModel
main_input_name = "hidden_states" main_input_name = "hidden_states"
uses_custom_attn_processor = True uses_custom_attn_processor = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment