From 5b9d53ea5546a2fbd6697d1320ec74da2599c591 Mon Sep 17 00:00:00 2001 From: zhihuiwan <15779896112@163.com> Date: Tue, 5 Mar 2024 10:46:06 +0800 Subject: [PATCH 1/2] support arbiter deepspeed engine Signed-off-by: zhihuiwan <15779896112@163.com> --- python/fate_flow/controller/task.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/fate_flow/controller/task.py b/python/fate_flow/controller/task.py index 3c3453f2f..f87fef197 100644 --- a/python/fate_flow/controller/task.py +++ b/python/fate_flow/controller/task.py @@ -120,8 +120,8 @@ def update_local(task): @staticmethod def update_launcher_config(task, launcher_name, task_parameters): # support deepspeed and other launcher - if task.f_role == "arbiter": - return + # if task.f_role == "arbiter": + # return schedule_logger(task.f_job_id).info(f"task runtime launcher name: {launcher_name}") if launcher_name and launcher_name != LauncherType.DEFAULT: task_parameters.launcher_name = task.f_launcher_name = launcher_name From b7909e8f02c37b0db24f26a305a97c80c6590555 Mon Sep 17 00:00:00 2001 From: zhihuiwan <15779896112@163.com> Date: Tue, 5 Mar 2024 10:48:02 +0800 Subject: [PATCH 2/2] update examples Signed-off-by: zhihuiwan <15779896112@163.com> --- examples/lr/train_lr.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/lr/train_lr.yaml b/examples/lr/train_lr.yaml index 1b13597d7..ccd59ffe9 100644 --- a/examples/lr/train_lr.yaml +++ b/examples/lr/train_lr.yaml @@ -80,7 +80,7 @@ dag: - lr_0 inputs: data: - input_data: + input_datas: task_output_artifact: - output_artifact_key: train_output_data producer_task: lr_0