Skip to content

z-image support npu#13689

Open
luren55 wants to merge 2 commits intohuggingface:mainfrom
Zhibin-Mo:main
Open

z-image support npu#13689
luren55 wants to merge 2 commits intohuggingface:mainfrom
Zhibin-Mo:main

Conversation

@luren55
Copy link
Copy Markdown

@luren55 luren55 commented May 7, 2026

What does this PR do?

Notice: Here is a PR extraction from #12979 to support z-image runs on NPU.

Issue: The Diffusers Z-Image pipeline fails on Ascend NPU because aclnnIndex lacks support for complex64. When freqs_cis is stored as a complex tensor on the NPU and then indexed, the call crashes.

Fix: Rework the RoPE frequency handling so that the NPU never needs to index complex64 tensors.

Running Environment:
I run this model base on cache-dit(https://github.com/vipshop/cache-dit/) which depends on diffusers to inference.

Command:
python3 generate.py zimage --model-path /home/weights/Z-Image-Turbo --attn _native_npu

Error before fixed:
image

Result after fixed:
image

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@github-actions github-actions Bot added models size/M PR with diff < 200 LOC labels May 7, 2026
@luren55
Copy link
Copy Markdown
Author

luren55 commented May 7, 2026

@yiyixuxu Hello, here is an extracion from the previous PR(#12979), can you review this PR? Thank you.

@luren55 luren55 mentioned this pull request May 8, 2026
6 tasks
@github-actions github-actions Bot added size/S PR with diff < 50 LOC and removed size/M PR with diff < 200 LOC labels May 8, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

models size/S PR with diff < 50 LOC

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant