Fix dtype model loading (#1449)

Commit

2 years ago

Fix dtype model loading (#1449) * Add test * up * no bfloat16 for mps * fix * rename test