Skip to content

Commit f12dc97

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

File tree

1 file changed

+1
-21
lines changed

1 file changed

+1
-21
lines changed

tests/models/transformers/test_models_transformer_ltx.py

Lines changed: 1 addition & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -19,13 +19,7 @@
1919

2020
from diffusers import LTXVideoTransformer3DModel
2121
from diffusers.utils.testing_utils import (
22-
enable_full_determinism,
23-
is_torch_compile,
24-
require_torch_2,
25-
require_torch_gpu,
26-
slow,
27-
torch_device,
28-
)
22+
from diffusers.utils.testing_utils import enable_full_determinism, torch_device
2923

3024
from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin
3125

@@ -87,20 +81,6 @@ def prepare_init_args_and_inputs_for_common(self):
8781

8882
def test_gradient_checkpointing_is_applied(self):
8983
expected_set = {"LTXVideoTransformer3DModel"}
90-
super().test_gradient_checkpointing_is_applied(expected_set=expected_set)
91-
92-
@require_torch_gpu
93-
@require_torch_2
94-
@is_torch_compile
95-
@slow
96-
def test_torch_compile_recompilation_and_graph_break(self):
9784
torch._dynamo.reset()
98-
init_dict, inputs_dict = self.prepare_init_args_and_inputs_for_common()
99-
100-
model = self.model_class(**init_dict).to(torch_device)
101-
model.eval()
102-
model = torch.compile(model, fullgraph=True)
103-
104-
with torch._dynamo.config.patch(error_on_recompile=True), torch.no_grad():
10585
_ = model(**inputs_dict)
10686
_ = model(**inputs_dict)

0 commit comments

Comments
 (0)