ColossalAI/colossalai/inference/core
Gao, Ruiyuan e9032fb0b2
[colossalai/checkpoint_io/...] fix bug in load_state_dict_into_model; format error msg (#6020)
* 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>
2024-09-02 16:56:35 +08:00
..
__init__.py [doc] updated inference readme (#5343) 2024-02-02 14:31:10 +08:00
async_engine.py [Inference] Fix API server, test and example (#5712) 2024-05-15 15:47:31 +08:00
base_engine.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
diffusion_engine.py [Feat] Distrifusion Acceleration Support for Diffusion Inference (#5895) 2024-07-30 10:43:26 +08:00
engine.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
llm_engine.py [HotFix] CI,import,requirements-test for #5838 (#5892) 2024-07-08 22:32:06 +08:00
plugin.py [colossalai/checkpoint_io/...] fix bug in load_state_dict_into_model; format error msg (#6020) 2024-09-02 16:56:35 +08:00
request_handler.py [Feat] Diffusion Model(PixArtAlpha/StableDiffusion3) Support (#5838) 2024-07-08 16:02:07 +08:00
rpc_engine.py [pre-commit.ci] pre-commit autoupdate (#5572) 2024-07-01 17:16:41 +08:00