device = finetuning_args.device (#1379)

DongBaiYue · web-flow · commit 6357e913103a · 2025-12-01T10:33:16.000+08:00
* text_sft_reader默认设置GPU设备，XPU需要传入device参数。
diff --git a/erniekit/train/vl_sft/workflow.py b/erniekit/train/vl_sft/workflow.py
@@ -700,6 +700,7 @@ def compute_metrics(p):
                 "use_train_part_sharding": finetuning_args.text_use_train_part_sharding,
                 "rope_3d": model_args.rope_3d,
                 "chat_template": preprocess_args.chat_template,
+                "device": finetuning_args.device,
             }
 
             text_sft_train_reader = create_pyreader(config_dataset_text)

Original file line number	Diff line number	Diff line change
`@@ -700,6 +700,7 @@ def compute_metrics(p):`
`700`	`700`	`"use_train_part_sharding": finetuning_args.text_use_train_part_sharding,`
`701`	`701`	`"rope_3d": model_args.rope_3d,`
`702`	`702`	`"chat_template": preprocess_args.chat_template,`
	`703`	`+ "device": finetuning_args.device,`
`703`	`704`	`}`
`704`	`705`
`705`	`706`	`text_sft_train_reader = create_pyreader(config_dataset_text)`