# pipeline model
model = get_model(model_config, ds_args, activation_checkpointing_config)
engine, _, _, _ = deepspeed.initialize(
ds_args,
model=model,
model_parameters=[p for p in model.parameters() if p.requires_grad]
)
# use `convert2ckpt.py`
engine.load_checkpoint(model_args.init_ckpt, load_module_only=True)
嗨,帅哥,你这边的工程很棒,我在学习的工程中有一些疑问,希望你能抽空解答一下。具体问提如下:
engine.load_checkpoint这里是不是必须加载ckpt,hf格式的为啥不可以呢?