diffusers
feat(training-utils): support device and dtype params in compute_density_for_timestep_sampling
#10699
Merged
Merged
feat(training-utils): support device and dtype params in compute_density_for_timestep_sampling #10699
sayakpaul
merged
4 commits into
huggingface:main
from
badayvedat:feat/add-device-tpype
feat(training-utils): support device and dtype params in compute_dens…
3667b3b5
chore: update type hint
8c8ada47
a-r-r-o-w
approved these changes
on 2025-02-01
refactor: use union for type hint
ba039509
sayakpaul
approved these changes
on 2025-02-01
Merge branch 'main' into feat/add-device-tpype
b92d43f6
Login to write a write a comment.
Login via GitHub