"git@developer.sourcefind.cn:OpenDAS/bitsandbytes.git" did not exist on "49c044b1daa22f17af6907ad26ae604758d55e1a"
Unverified Commit 19085ac8 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

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

u[
parent 041501ae
...@@ -1793,11 +1793,6 @@ class ModelTesterMixin: ...@@ -1793,11 +1793,6 @@ class ModelTesterMixin:
if not self.model_class._supports_group_offloading: if not self.model_class._supports_group_offloading:
pytest.skip("Model does not support 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): def _has_generator_arg(model):
sig = inspect.signature(model.forward) sig = inspect.signature(model.forward)
params = sig.parameters params = sig.parameters
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment