fix(llm): 修复微调保存步数默认值逻辑

- 使用 Optional.ofNullable() 处理 fineTuningTask.getSaveSteps() 的值
- 在原值基础上加 1,确保保存步数至少为 1
- 优化了代码的可读性和健壮性
This commit is contained in:
Liuyang 2025-02-26 11:48:08 +08:00
parent 012a04b860
commit 3be8b1e869

View File

@ -177,7 +177,9 @@ 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());
int saveStep = Optional.ofNullable(fineTuningTask.getSaveSteps())
.orElse(0) + 1;
req.setTrainEpoch(saveStep);
req.setSuffix(fineTuningTask.getTaskName());
req.setRemark(fineTuningTask.getTaskIntro());
req.setTrainBatchSize(fineTuningTask.getBatchSize());