From e4f405dad2e194ff2ea8cdf182d4daef57614617 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Wed, 24 Sep 2025 15:17:10 +0530 Subject: [PATCH] u[ --- tests/models/test_modeling_common.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py index 5e7be62342c3..20989eeab47f 100644 --- a/tests/models/test_modeling_common.py +++ b/tests/models/test_modeling_common.py @@ -1794,11 +1794,6 @@ def test_group_offloading_with_disk(self, offload_type, record_stream, atol=1e-5 if not self.model_class._supports_group_offloading: pytest.skip("Model does not support group offloading.") - if self.model_class.__name__ == "QwenImageTransformer2DModel": - pytest.skip( - "QwenImageTransformer2DModel doesn't support group offloading with disk. Needs to be investigated." - ) - def _has_generator_arg(model): sig = inspect.signature(model.forward) params = sig.parameters