Keep parameter dtype through ZeRO-3 weight quantization (#8215)