Commit bb41bd2a authored by hepj987's avatar hepj987
Browse files

修改说明

parent be9eaaf6
......@@ -158,7 +158,7 @@ pip install deepspeed-0.7.3+unknown-cp37-cp37m-linux_x86_64.whl -i http://pypi.t
install_cuda_version = torch_info['cuda_version']
install_hip_version = torch_info['hip_version']
修改
修改/usr/local/lib/python3.7/site-packages/deepspeed/runtime/state_dict_factory.py文件
第177行def check_ckpt_list(self):函数
删除mp_world_size判断
if 'mp_world_size' in sd.keys():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment