From 1944365db7f23671d619e11941c08f16e62160c6 Mon Sep 17 00:00:00 2001
From: leon <800222@163.com>
Date: Wed, 26 Feb 2025 11:17:40 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9EsaveSteps?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java   | 4 ++++
 .../module/llm/service/async/AsyncFineTuningTaskService.java  | 1 +
 2 files changed, 5 insertions(+)

diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
index ebc6211e8..83e9b9ea8 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/finetuningtask/FineTuningTaskDO.java
@@ -78,6 +78,10 @@ public class FineTuningTaskDO extends BaseDO {
      * 训练轮数
      */
     private Integer epoch;
+    /**
+     * 训练步长
+     */
+    private Integer saveSteps;
     /**
      * 截断长度
      */
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
index 6ae9b1260..11e5a6a14 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
@@ -177,6 +177,7 @@ public class AsyncFineTuningTaskService {
         AigcFineTuningCreateReqVO req = new AigcFineTuningCreateReqVO();
         req.setModel(fineTuningTask.getBaseModel());
         req.setTrainEpoch((fineTuningTask.getEpoch() == 0) ? 1 : fineTuningTask.getEpoch());
+        req.setTrainEpoch((fineTuningTask.getSaveSteps() == 0) ? 1 : fineTuningTask.getSaveSteps());
         req.setSuffix(fineTuningTask.getTaskName());
         req.setRemark(fineTuningTask.getTaskIntro());
         req.setTrainBatchSize(fineTuningTask.getBatchSize());