"examples/pytorch/image-pretraining/run_mim.py" did not exist on "6c7b68d414cd170c0efd3b745a924cbef6e2a5d8"
-
Wang, Yi authored
* [run_clm example] add torch_dtype option for model load. for BLOOM 175B model. peak memory will reduce about 350G for inference. the weight of BLOOM in model hub is bfloat16 Signed-off-by:
Wang, Yi A <yi.a.wang@intel.com> * add other type in option * fix style Signed-off-by:
9c9fe89f
