[BUG]deepspeed_init() got an unexpected keyword argument 'resume_from_checkpoint'

[Yummy813]

Traceback (most recent call last):
File "/LMFlow/examples/raft_align.py", line 217, in
File "/LMFlow/examples/raft_align.py", line 217, in
main() main()

File "/LMFlow/examples/raft_align.py", line 208, in main
File "/LMFlow/examples/raft_align.py", line 208, in main
aligned_model = aligner.align(
File "/LMFlow/src/lmflow/pipeline/raft_aligner.py", line 616, in align
aligned_model = aligner.align(
File "/LMFlow/src/lmflow/pipeline/raft_aligner.py", line 616, in align
raft_trainer.train(resume_from_checkpoint=False, is_first_time=True)
File "/LMFlow/src/lmflow/pipeline/utils/raft_trainer.py", line 1600, in train
raft_trainer.train(resume_from_checkpoint=False, is_first_time=True)
File "/LMFlow/src/lmflow/pipeline/utils/raft_trainer.py", line 1600, in train
return inner_training_loop1(
File "/LMFlow/src/lmflow/pipeline/utils/raft_trainer.py", line 2021, in _inner_training_loop
return inner_training_loop1(
File "/LMFlow/src/lmflow/pipeline/utils/raft_trainer.py", line 2021, in _inner_training_loop
deepspeed_engine, optimizer, lr_scheduler = deepspeed_init(
TypeError: deepspeed_init() got an unexpected keyword argument 'resume_from_checkpoint'
deepspeed_engine, optimizer, lr_scheduler = deepspeed_init(
TypeError: deepspeed_init() got an unexpected keyword argument 'resume_from_checkpoint'