src/diffusers/models/unet_2d_condition_flax.py · 038b42db942e724c86ede866465048fd57e08ee2 · renzhc / diffusers_dcu

Improve docs and type hints (#5759) · 038b42db

Aryan V S authored Nov 16, 2023



* improvement: docs and type hints

* improvement: docs and type hints

minor refactor

* improvement: docs and type hints

* update with suggestions from review
Co-Authored-By: Dhruv Nair <dhruv.nair@gmail.com>

---------
Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

038b42db

unet_2d_condition_flax.py 21.1 KB

Replace unet_2d_condition_flax.py