reorder model wrap + bug fix (#1799)

Commit

3 years ago

reorder model wrap + bug fix (#1799) * reorder model wrap * bug fix Co-authored-by: Prathik Rao <prathikrao@microsoft.com>