Fix dtype bug when offload_state_dict=True and dtype is specified
#3721
| Job | Run time |
|---|---|
| 1m 40s | |
| 1m 40s |
offload_state_dict=True and dtype is specified
#3721
| Job | Run time |
|---|---|
| 1m 40s | |
| 1m 40s |