Closed niw closed 2 weeks ago
On mps, it only supports float16 for autocast. Use float16 instead of bfloat16 on mps for now.
float16
autocast
bfloat16
This patch also use dtype for vae as well so that it can use bfloat16 or float16 on mps and other environment.
dtype
On mps, it only supports
float16forautocast. Usefloat16instead ofbfloat16on mps for now.This patch also use
dtypefor vae as well so that it can usebfloat16orfloat16on mps and other environment.