The remote pipeline uses flaot64 / doubles breaks the pipeline on MPS and NPU/XPU's
#5
by Vargol - opened
Are the uses of float64 on line 236
https://huggingface.co/OpenMOSS-Team/MOSS-Image-SFT/blob/main/pipeline_moss_image.py#L236
and the double() call on the next line, 237,
really necessary ? MacOS's Metal/MPS and many XPU NPU's do not support float64 and thus the pipeline breaks.
I'm currently yarding a version where the float64 has been modified to float32, and the double() call removed but my Mac is really slow so will take some time to do 50 steps 😃
Seems to work okay.
