src/diffusers/schedulers/scheduling_pndm_flax.py · 0a09af2f0a55cf96e0a3ad87d7dc55a6e958fd77 · renzhc / diffusers_dcu

Suraj Patil authored Oct 13, 2022

* update flax scheduler API

* remoev set format

* fix call to scale_model_input

* update flax pndm

* use int32

* update docstr

0a09af2f

scheduling_pndm_flax.py 21.9 KB