Skip to content

Add Support for Z-Image Series #12703

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
yiyixuxu merged 29 commits into from
Nov 25, 2025
+1,382 −0
Merged

Add Support for Z-Image Series #12703

Changes from 1 commit
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
42658fa
Add Support for Z-Image.
JerryWu-code Nov 23, 2025
3e74bb2
Reformatting with make style, black & isort.
JerryWu-code Nov 23, 2025
a4b89a0
Remove init, Modify import utils, Merge forward in transformers block…
JerryWu-code Nov 24, 2025
7df350d
modified main model forward, freqs_cis left
ChrisLiu6 Nov 24, 2025
1dd587b
Merge remote-tracking branch 'JerryWu-code/z-image-dev' into fork/Jer…
ChrisLiu6 Nov 24, 2025
aae03cf
refactored to add B dim
ChrisLiu6 Nov 24, 2025
21d8130
fixed stack issue
ChrisLiu6 Nov 24, 2025
e3dfa9e
fixed modulation bug
ChrisLiu6 Nov 24, 2025
a7fa731
fixed modulation bug
ChrisLiu6 Nov 24, 2025
1e0cefe
fix bug
ChrisLiu6 Nov 24, 2025
7adaae8
remove value_from_time_aware_config
ChrisLiu6 Nov 24, 2025
5b4c907
styling
ChrisLiu6 Nov 24, 2025
2bb39f4
Fix neg embed and devide / bug; Reuse pad zero tensor; Turn cat -> re…
JerryWu-code Nov 24, 2025
71e8049
Replace padding with pad_sequence; Add gradient checkpointing.
JerryWu-code Nov 24, 2025
fbf26b7
Fix flash_attn3 in dispatch attn backend by _flash_attn_forward, repl…
JerryWu-code Nov 24, 2025
6c0c059
Fix Docstring and Make Style.
JerryWu-code Nov 24, 2025
28685dd
Revert "Fix flash_attn3 in dispatch attn backend by _flash_attn_forwa…
ChrisLiu6 Nov 25, 2025
8e391b7
update z-image docstring
ChrisLiu6 Nov 25, 2025
3b22e84
Revert attention dispatcher
ChrisLiu6 Nov 25, 2025
3d1a7aa
update z-image docstring
ChrisLiu6 Nov 25, 2025
336c5ce
styling
ChrisLiu6 Nov 25, 2025
38a89ed
Recover attention_dispatch.py with its origin impl, later would speci…
JerryWu-code Nov 25, 2025
69d61e5
Fix prev bug, and support for prompt_embeds pass in args after prompt…
JerryWu-code Nov 25, 2025
549ad57
Merge branch 'z-image-dev-ql' into z-image-dev
JerryWu-code Nov 25, 2025
1dd8f3c
Remove einop dependency.
JerryWu-code Nov 25, 2025
2f2d8c3
Merge branch 'z-image-dev' into z-image
JerryWu-code Nov 25, 2025
a74a0c4
Merge remote-tracking branch 'origin/main' into z-image
JerryWu-code Nov 25, 2025
e49a1f9
remove redundant imports & make fix-copies
ChrisLiu6 Nov 25, 2025
1048d0a
fix import
ChrisLiu6 Nov 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
refactored to add B dim
  • Loading branch information
@ChrisLiu6
ChrisLiu6 committed Nov 24, 2025
commit aae03cf6a6f6d1562815f98b50521ec2f7d7f07e
Loading