Fix regression loading dtype (#34409)

Commit

1 year ago

Fix regression loading dtype (#34409) * fix regression * add test for torchao * expected output * better fix