diffusers

feat(training-utils): support device and dtype params in compute_density_for_timestep_sampling

#10699

Merged