Use fp16 if checkpoint weights are fp16 and the model supports it.

Use fp16 if checkpoint weights are fp16 and the model supports it. #608