[validation] Apply UNet3D class conditioning

[lindicaphxag-tech]

Validation PR for the focused fix for huggingface#7053. Makes the documented class_labels argument effective through an optional class embedding without changing existing checkpoint configs.

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}