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

sayakpaul merged 4 commits intohuggingface:mainhuggingface/diffusers:mainfrom

badayvedat:feat/add-device-tpypebadayvedat/diffusers:feat/add-device-tpypeCopy head branch name to clipboard