send model to the correct device (#18800)

Commit

3 years ago

send model to the correct device (#18800) Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>