PR #12857 Add z-image-omni-base implementation

Add z-image-omni-base implementation #12857

yiyixuxu merged 13 commits into huggingface:main from JerryWu-code:z-image-omni-base

Add z-image-omni-base implementation

3435ba2a

yiyixuxu commented on 2025-12-18

Merged into one transformer for Z-Image.

3cbb38d9

Merge branch 'main' into z-image-omni-base

3e60fa7b

Fix bugs for controlnet after merging the main branch new feature.

9180579b

Fix for auto_pipeline, Add Styling.

4c14cf3d

Refactor noise handling and modulation

5bc676c0

JerryWu-code force pushed from 70bc2c87 to 5bc676c0 101 days ago

Styling & Formatting.

732c5275

Rewrite code with more non-forward func & clean forward.

5df53964

Styling & Formatting.

0c53a381

Merge branch 'main' into z-image-omni-base

4dec9b52

yiyixuxu commented on 2025-12-23

yiyixuxu approved these changes on 2025-12-23

Manual check fix-copies in controlnet, Add select_per_token, _patchif…

969d8d09

Add Import in pipeline __init__.py.

9189019e

Merge branch 'main' into z-image-omni-base

bf19a3a8

yiyixuxu merged f6b6a718 into main 97 days ago

yiyixuxu commented on 2025-12-24

Reviewers

yiyixuxu

JerryWu-code

Assignees

No one assigned

Labels

None yet

Milestone

No milestone