fix(unets): preserve scalar float timestep dtype in UNet1D/2D + FlaxUNet2DCondition (#13654)#13669

Open

Anai-Guo wants to merge 3 commits intohuggingface:mainfrom

Anai-Guo:fix-unet-1d-2d-flax-float-timestep

Commits on May 1, 2026

fix(unets): preserve scalar float timestep dtype in UNet1DModel
Anai-Guo
committed
fix(unets): preserve scalar float timestep dtype in UNet2DModel
Anai-Guo
committed
fix(unets): preserve scalar float timestep dtype in FlaxUNet2DConditionModel
Anai-Guo
committed