Compare commits

...

2 Commits

Author SHA1 Message Date
Sayak Paul
8a0a2793af Merge branch 'main' into qwen-go-disk 2025-09-24 15:18:24 +05:30
sayakpaul
e4f405dad2 u[ 2025-09-24 15:17:10 +05:30

View File

@@ -1794,11 +1794,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