Skip to content

Commit 19085ac

Browse files
authored
Don't skip Qwen model tests for group offloading with disk (huggingface#12382)
u[
1 parent 041501a commit 19085ac

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

tests/models/test_modeling_common.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1793,11 +1793,6 @@ def test_group_offloading_with_disk(self, offload_type, record_stream, atol=1e-5
17931793
if not self.model_class._supports_group_offloading:
17941794
pytest.skip("Model does not support group offloading.")
17951795

1796-
if self.model_class.__name__ == "QwenImageTransformer2DModel":
1797-
pytest.skip(
1798-
"QwenImageTransformer2DModel doesn't support group offloading with disk. Needs to be investigated."
1799-
)
1800-
18011796
def _has_generator_arg(model):
18021797
sig = inspect.signature(model.forward)
18031798
params = sig.parameters

0 commit comments

Comments
 (0)