diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java index 370491db1..9c377f1ec 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskPageReqVO.java @@ -1,6 +1,8 @@ package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import io.swagger.v3.oas.annotations.media.Schema; import cn.iocoder.yudao.framework.common.pojo.PageParam; @@ -43,7 +45,7 @@ public class FineTuningTaskPageReqVO extends PageParam { private Long dataset; @Schema(description = "学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") private Integer epoch; @@ -94,4 +96,4 @@ public class FineTuningTaskPageReqVO extends PageParam { @Schema(description ="基础模型服务Id") private Long baseModelId; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java index eb65f0402..93fa2ccc9 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskRespVO.java @@ -4,6 +4,8 @@ import cn.iocoder.yudao.module.llm.service.http.vo.AigcFineTuningDetailRespVO; import com.alibaba.fastjson.JSONObject; import io.swagger.v3.oas.annotations.media.Schema; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import org.springframework.format.annotation.DateTimeFormat; import java.time.LocalDateTime; @@ -58,7 +60,7 @@ public class FineTuningTaskRespVO { @Schema(description = "学习率") @ExcelProperty("学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") @ExcelProperty("训练轮数") diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java index 4451f79f8..5bb1d506c 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/finetuningtask/vo/FineTuningTaskSaveReqVO.java @@ -2,6 +2,8 @@ package cn.iocoder.yudao.module.llm.controller.admin.finetuningtask.vo; import io.swagger.v3.oas.annotations.media.Schema; import lombok.*; + +import java.math.BigDecimal; import java.util.*; import javax.validation.constraints.*; @@ -41,7 +43,7 @@ public class FineTuningTaskSaveReqVO { private Long dataset; @Schema(description = "学习率") - private Integer learningRate; + private BigDecimal learningRate; @Schema(description = "训练轮数") private Integer epoch; @@ -89,4 +91,4 @@ public class FineTuningTaskSaveReqVO { @Schema(description = "基础模型服务Id") private Long baseModelId; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java index e868db056..c5b29eac0 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java @@ -154,6 +154,7 @@ public class TrainHttpService { // TODO: 在上个方法中已经将数据集的文件id赋予,调试时需要写死再放开 // String fileId = "6237ed4d-a046-479c-80d6-8579a0283994"; // req.setFileId(fileId); + log.info(" finetuningCreate:{}", JSON.toJSONString(req)); String res = HttpUtils.post(llmBackendProperties.getFinetuningCreate(), headers, JSON.toJSONString(req)); log.info(" finetuningCreate:{}", res); AigcRespVO aigcRespVO = JSON.parseObject(res, AigcRespVO.class);