From 72ce96ae5ab31e27fa5443beafbfb6bf4bfca3a6 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 20 Oct 2025 19:48:50 -1000 Subject: [PATCH] up --- .../transformers/test_models_transformer_skyreels_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/transformers/test_models_transformer_skyreels_v2.py b/tests/models/transformers/test_models_transformer_skyreels_v2.py index 8c36d8256e..75198d13cb 100644 --- a/tests/models/transformers/test_models_transformer_skyreels_v2.py +++ b/tests/models/transformers/test_models_transformer_skyreels_v2.py @@ -22,13 +22,13 @@ from ...testing_utils import ( enable_full_determinism, torch_device, ) -from ..test_modeling_common import ModelTesterMixin, TorchCompileTesterMixin +from ..test_modeling_common import ModelTesterMixin enable_full_determinism() -class SkyReelsV2Transformer3DTests(ModelTesterMixin, TorchCompileTesterMixin, unittest.TestCase): +class SkyReelsV2Transformer3DTests(ModelTesterMixin, unittest.TestCase): model_class = SkyReelsV2Transformer3DModel main_input_name = "hidden_states" uses_custom_attn_processor = True