mirror of https://github.com/hpcaitech/ColossalAI
e9032fb0b2
* fix bug in load_state_dict_into_model; format error msg * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update utils.py to support checking missing_keys * Update general_checkpoint_io.py fix bug in missing_keys error message * retrigger tests --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> |
||
---|---|---|
.. | ||
__init__.py | ||
checkpoint_io_base.py | ||
general_checkpoint_io.py | ||
hybrid_parallel_checkpoint_io.py | ||
index_file.py | ||
moe_checkpoint.py | ||
utils.py |