Skip to content
Merged
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
fixed modulation bug
  • Loading branch information
ChrisLiu6 committed Nov 24, 2025
commit a7fa73140c61e8926cdb7704e5d83cfabb080330
5 changes: 0 additions & 5 deletions src/diffusers/utils/import_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -230,7 +230,6 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[b
_aiter_available, _aiter_version = _is_package_available("aiter")
_kornia_available, _kornia_version = _is_package_available("kornia")
_nvidia_modelopt_available, _nvidia_modelopt_version = _is_package_available("modelopt", get_dist_name=True)
_apex_available, _apex_version = _is_package_available("apex")


def is_torch_available():
Expand Down Expand Up @@ -421,10 +420,6 @@ def is_kornia_available():
return _kornia_available


def is_apex_available():
return _apex_available


# docstyle-ignore
FLAX_IMPORT_ERROR = """
{0} requires the FLAX library but it was not found in your environment. Checkout the instructions on the
Expand Down