RulinShao / LightSeq

Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers
179 stars 8 forks source link

Fix outdated package name #4

Closed 66RING closed 8 months ago

66RING commented 8 months ago

lasync_communication and lightseq_ckpt_monkey_patch.replace_hf_ckpt_with_fast_ckpt not found.

rename to

from async_communication import

and

from lightseq_ckpt_monkey_patch import replace_hf_ckpt_with_new_ckpt, clear_all_buffers_at_the_end_of_training
replace_hf_ckpt_with_new_ckpt()