diff --git a/configs/7B_sft.py b/configs/7B_sft.py index 0c37495..9a866dd 100644 --- a/configs/7B_sft.py +++ b/configs/7B_sft.py @@ -150,8 +150,8 @@ pipeline parallel (dict): tensor parallel: tensor parallel size, usually the number of GPUs per node. """ parallel = dict( - zero1=-1, - pipeline=dict(size=2, interleaved_overlap=True), + zero1=8, + pipeline=dict(size=1, interleaved_overlap=True), sequence_parallel=False, )