Fix k_dpm_2 & k_dpm_2_a on MPS (#2241)

Commit

2 years ago

Fix k_dpm_2 & k_dpm_2_a on MPS (#2241) Needed to convert `timesteps` to `float32` a bit sooner. Fixes #1537