refactor(llm): 重构模型微调相关代码

- 将 AigcFineTuningCreateReqVO 和 FineTuningTaskDO 中的 learningRate 字段类型从 double/Integer 改为 BigDecimal,以提高精度
- 优化 ModelService 中的 createFineTuningTask 方法,异常处理改为抛出 RuntimeException
This commit is contained in:
Liuyang 2025-02-14 10:03:20 +08:00
parent 08c82dbb83
commit 1f20747f9c
3 changed files with 7 additions and 3 deletions

View File

@ -1,6 +1,8 @@
package cn.iocoder.yudao.module.llm.dal.dataobject.finetuningtask;
import lombok.*;
import java.math.BigDecimal;
import java.util.*;
import java.time.LocalDateTime;
import java.time.LocalDateTime;
@ -71,7 +73,7 @@ public class FineTuningTaskDO extends BaseDO {
/**
* 学习率
*/
private Integer learningRate;
private BigDecimal learningRate;
/**
* 训练轮数
*/

View File

@ -90,7 +90,7 @@ public class ModelService {
}
return null;
} catch (Exception e) {
return null;
throw new RuntimeException(e);
}
}

View File

@ -4,6 +4,8 @@ import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.math.BigDecimal;
@Data
@AllArgsConstructor
@NoArgsConstructor
@ -20,7 +22,7 @@ public class AigcFineTuningCreateReqVO {
private int evalBatchSize;
private int accumulationSteps;
private int procPerNode;
private double learningRate;
private BigDecimal learningRate;
private int modelMaxLength;
private boolean lora;
}