[New model] Qwen-Image-2.1 support (rebase + review follow-ups) (#8099)

Signed-off-by: NumberWan <wantszkin2003@gmail.com>
Signed-off-by: SamitHuang <285365963@qq.com>
Signed-off-by: Samit <285365963@qq.com>
Co-authored-by: Isotr0py <Isotr0py@outlook.com>
Co-authored-by: Tianyu Guo <guoty9@mail2.sysu.edu.cn>
Co-authored-by: Cursor <cursoragent@cursor.com>
Co-authored-by: SamitHuang <285365963@qq.com>
Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
This commit is contained in:
authored and GitHub committed 2026-10-07 21:23:59 +08:00
1 parent 8ef77be66e
commit 3dc35694b3
52 files changed
+8058 -136

No files matched your search

@@ -521,6 +521,35 @@ def test_compile_transformer_regionally_compiles_blocks(monkeypatch, use_hsdp):
]
@pytest.mark.core_model
@pytest.mark.cpu
def test_compile_transformer_scopes_inductor_cudagraphs_to_decode_graph(monkeypatch):
model = SimpleNamespace(enable_cuda_graph_decode=True)
runner = _make_compile_runner(model)
compile_calls = []
def _regionally_compile(target, *args, **kwargs):
compile_calls.append(kwargs)
return target
monkeypatch.setattr(model_runner_module, "regionally_compile", _regionally_compile)
import torch._inductor.config as inductor_config
before = inductor_config.triton.cudagraphs
DiffusionModelRunner._compile_transformer(runner, "transformer")
assert inductor_config.triton.cudagraphs == before
assert compile_calls == [
{
"dynamic": True,
"options": {
"triton.cudagraphs": False,
"triton.cudagraph_trees": False,
},
}
]
@pytest.mark.core_model
@pytest.mark.cpu
def test_compile_transformer_uses_regional_dynamic_false_config(monkeypatch):