Support unittest for Z-image ⚡️ #12715
yiyixuxu
merged
38 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
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
remove redundant imports & make fix-copies
e49a1f90
fix import
1048d0a9
Support for num_images_per_prompt>1; Remove redundant unquote variables.
266e169a
Fix bugs for num_images_per_prompt with actual batch.
12d2fb28
Add unit tests for Z-Image.
9a049f0c
Refine unitest and skip for cases needed separate test env; Fix compa…
c4e4a57e
Add clean env for test_save_load_float16 separ test; Add Note; Styling.
6f2808b6
Merge current branch into ours for next pr compatibility.
e48060ce
Merge branch 'main' into z-image
27a37cd4
Update dtype mentioned by yiyi.
aeed8904
Merge branch 'main' into z-image
e277137b
yiyixuxu
approved these changes
on 2025-11-26
yiyixuxu
commented
on 2025-11-26
Login to write a write a comment.
Login via GitHub