AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Don't skip Qwen model tests for group offloading with disk (#12382)

Browse Source 
u[
This commit is contained in:
Sayak Paul
2025-09-29 13:08:05 +05:30
committed by GitHub
parent 041501aea9
commit 19085ac8f4
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/models/test_modeling_common.py
View File

@@ -1793,11 +1793,6 @@ class ModelTesterMixin:
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