Fix DS Stable Diffusion for latest diffusers version (#4770)
This PR fixes the DeepSpeed `UNet` forward function to work with the
latest `diffusers` version.
Manual `nv-sd` workflow run:
https://github.com/microsoft/DeepSpeed/actions/runs/7092016979
Fixes #4760
