Add an option --fp16-unet to force using fp16 for the unet.

Add an option --fp16-unet to force using fp16 for the unet. #113