From 337237ee6690316bdb74937239cbc0771a1fbe6b Mon Sep 17 00:00:00 2001
From: leon <800222@163.com>
Date: Fri, 28 Feb 2025 15:18:53 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9saveSteps?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../module/llm/service/async/AsyncFineTuningTaskService.java  | 4 ++--
 .../module/llm/service/http/vo/AigcFineTuningCreateReqVO.java | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

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 aad2fed39..a951c0ea2 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,9 +177,9 @@ public class AsyncFineTuningTaskService {
         AigcFineTuningCreateReqVO req = new AigcFineTuningCreateReqVO();
         req.setModel(fineTuningTask.getBaseModel());
         req.setTrainEpoch((fineTuningTask.getEpoch() == 0) ? 1 : fineTuningTask.getEpoch());
-        int saveStep = Optional.ofNullable(fineTuningTask.getSaveSteps())
+        int saveSteps = Optional.ofNullable(fineTuningTask.getSaveSteps())
                                .orElse(0) + 1;
-        req.setTrainEpoch(saveStep);
+        req.setSaveSteps(saveSteps);
         req.setSuffix(fineTuningTask.getTaskName());
         req.setRemark(fineTuningTask.getTaskIntro());
         req.setTrainBatchSize(fineTuningTask.getBatchSize());
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
index 83741b186..c70cf0dcb 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
@@ -16,6 +16,7 @@ public class AigcFineTuningCreateReqVO {
         private String dataset;
         private String model;
         private int trainEpoch;
+        private int saveSteps;
         private String suffix;
         private String remark;
         private int trainBatchSize;