Error when run reshard_fsdp on opt-IML 30b for inference

Created by: zhanghaoie

Traceback (most recent call last): File "/opt/conda/envs/alpa/lib/python3.8/runpy.py", line 194, in _run_module_as_main return _run_code(code, main_globals, None, File "/opt/conda/envs/alpa/lib/python3.8/runpy.py", line 87, in _run_code exec(code, run_globals) File "/build/metaseq/metaseq/scripts/reshard_fsdp.py", line 245, in fire.Fire(reshard_fsdp_checkpoints) File "/opt/conda/envs/alpa/lib/python3.8/site-packages/fire/core.py", line 141, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) File "/opt/conda/envs/alpa/lib/python3.8/site-packages/fire/core.py", line 475, in _Fire component, remaining_args = _CallAndUpdateTrace( File "/opt/conda/envs/alpa/lib/python3.8/site-packages/fire/core.py", line 691, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) File "/build/metaseq/metaseq/scripts/reshard_fsdp.py", line 53, in reshard_fsdp_checkpoints resharded_state_dicts = reshard_fsdp_state_dicts( File "/build/metaseq/metaseq/scripts/reshard_fsdp.py", line 79, in reshard_fsdp_state_dicts shard_metadata=[s["shard_metadata"] for s in shard_state_dicts], File "/build/metaseq/metaseq/scripts/reshard_fsdp.py", line 79, in shard_metadata=[s["shard_metadata"] for s in shard_state_dicts], KeyError: 'shard_metadata'