[Bug] Please update transformer_zimage.py

[SatyrTai]

Checklist

• 1. I have searched for related issues and FAQs (https://nunchaku.tech/docs/nunchaku/faq/faq.html) but was unable to find a solution.
• 2. The issue persists in the latest version.
• 3. Please note that without environment information and a minimal reproducible example, it will be difficult for us to reproduce and address the issue, which may delay our response.
• 4. If your report is a question rather than a bug, please submit it as a discussion at https://github.com/mit-han-lab/nunchaku/discussions/new/choose. Otherwise, this issue will be closed.
• 5. If this is related to ComfyUI, please report it at https://github.com/mit-han-lab/ComfyUI-nunchaku/issues.
• 6. I will do my best to describe the issue in English.

Describe the Bug

Diffusers 0.37.0 was released two weeks ago.
They updated the "forward" method, which caused #895.
I hope this is helpful to you, and thank you for your work.

nunchaku/nunchaku/models/transformers/transformer_zimage.py

Line 352 in 8f41840

return super().forward(x, t, cap_feats, patch_size, f_patch_size, return_dict)

https://github.com/huggingface/diffusers/blob/v0.37.0/src/diffusers/models/transformers/transformer_z_image.py#L887

Environment

...

Reproduction Steps

...