Improve Wan docstrings (#11689)
* improve docstrings for wan
* Apply suggestions from code review
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
* make style
---------
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
