fix type-check for z-image transformer#12739

sayakpaul merged 3 commits intohuggingface:mainhuggingface/diffusers:mainfrom

xlite-dev:devxlite-dev/diffusers:devCopy head branch name to clipboard