Add Support for Z-Image Series #12703
yiyixuxu
merged
29 commits into
huggingface:main
from
JerryWu-code:z-image
Add Support for Z-Image.
42658fad
Reformatting with make style, black & isort.
3e74bb25
Remove init, Modify import utils, Merge forward in transformers block…
a4b89a08
modified main model forward, freqs_cis left
7df350d0
Merge remote-tracking branch 'JerryWu-code/z-image-dev' into fork/Jer…
1dd587bb
refactored to add B dim
aae03cf6
fixed stack issue
21d81302
fixed modulation bug
e3dfa9e6
fixed modulation bug
a7fa7314
fix bug
1e0cefe1
remove value_from_time_aware_config
7adaae88
styling
5b4c9074
Fix neg embed and devide / bug; Reuse pad zero tensor; Turn cat -> re…
2bb39f46
Replace padding with pad_sequence; Add gradient checkpointing.
71e8049a
Fix flash_attn3 in dispatch attn backend by _flash_attn_forward, repl…
fbf26b7e
Fix Docstring and Make Style.
6c0c059f
yiyixuxu
commented
on 2025-11-24
yiyixuxu
commented
on 2025-11-24
Revert "Fix flash_attn3 in dispatch attn backend by _flash_attn_forwa…
28685dd3
update z-image docstring
8e391b77
Revert attention dispatcher
3b22e84b
update z-image docstring
3d1a7aa3
styling
336c5cee
Recover attention_dispatch.py with its origin impl, later would speci…
38a89ede
Fix prev bug, and support for prompt_embeds pass in args after prompt…
69d61e50
Merge branch 'z-image-dev-ql' into z-image-dev
549ad576
Remove einop dependency.
1dd8f3cf
Merge branch 'z-image-dev' into z-image
2f2d8c3a
Merge remote-tracking branch 'origin/main' into z-image
a74a0c43
yiyixuxu
approved these changes
on 2025-11-25
yiyixuxu
commented
on 2025-11-25
remove redundant imports & make fix-copies
e49a1f90
fix import
1048d0a9
Login to write a write a comment.
Login via GitHub