[diffusion] platform: support WAN/FLUX/Qwen-Image/Qwen-Image-edit on Ascend (#13662)

Co-authored-by: dhx98 <haox.dai@gmail.com>
Co-authored-by: DHX98 <haoxiand@andrew.cmu.edu>
Co-authored-by: ronnie_zheng <zl19940307@163.com>
Co-authored-by: DHX98 <DHX98@noreply.gitcode.com>
Co-authored-by: Yuhao Yang <47235274+yhyang201@users.noreply.github.com>
This commit is contained in:
Makcum888e
2026-02-08 10:45:30 +08:00
committed by GitHub
co-authored by dhx98 DHX98 ronnie_zheng DHX98 Yuhao Yang
parent 7b83659310
commit 00248d85c7
25 changed files with 476 additions and 30 deletions
@@ -248,7 +248,11 @@ def init_distributed_environment(
# For MPS and MUSA, don't pass device_id as it doesn't support device indices
extra_args = (
{}
if (current_platform.is_mps() or current_platform.is_musa())
if (
current_platform.is_mps()
or current_platform.is_musa()
or current_platform.is_npu()
)
else dict(device_id=device_id)
)
@@ -618,6 +622,7 @@ def maybe_init_distributed_environment_and_model_parallel(
local_rank=local_rank,
distributed_init_method=distributed_init_method,
device_id=device,
backend=current_platform.get_torch_distributed_backend_str(),
timeout=dist_timeout,
)
initialize_model_parallel(