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 f620cc157..b9f14ad23 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 @@ -133,7 +133,7 @@ public class AsyncFineTuningTaskService { updateObj.setStatus(FinetuningTaskStatusEnum.WAITING.getStatus()); updateObj.setJobModelName(resp.getFineTunedModel()); updateObj.setTrainLog(resp.getTrainLog()); - log.info("微调任务创建成功。任务ID: {}, 任务模型名称: {} , 任务状态: {}", fineTuningTask.getId(), fineTuningTask.getJobModelName(),FinetuningTaskStatusEnum.WAITING.getStatus()); + log.info("微调任务创建成功。任务ID: {}, 任务模型名称: {} , 任务状态: {}", fineTuningTask.getId(), fineTuningTask.getJobModelName(), FinetuningTaskStatusEnum.WAITING.getStatus()); } else { updateObj.setStatus(FinetuningTaskStatusEnum.FAILED.getStatus()); log.error("微调任务创建失败。任务ID: {}", fineTuningTask.getId()); @@ -146,7 +146,7 @@ public class AsyncFineTuningTaskService { } catch (Exception e) { log.error("创建微调任务时发生异常。任务ID: {}", fineTuningTask.getId(), e); - e.printStackTrace(); + throw e; } } @@ -178,7 +178,7 @@ public class AsyncFineTuningTaskService { req.setModel(fineTuningTask.getBaseModel()); req.setTrainEpoch((fineTuningTask.getEpoch() == 0) ? 1 : fineTuningTask.getEpoch()); int saveStep = Optional.ofNullable(fineTuningTask.getSaveSteps()) - .orElse(0) + 1; + .orElse(0) + 1; req.setTrainEpoch(saveStep); req.setSuffix(fineTuningTask.getTaskName()); req.setRemark(fineTuningTask.getTaskIntro()); @@ -196,7 +196,7 @@ public class AsyncFineTuningTaskService { @Async public void startFineTuningTask (FineTuningTaskDO fineTuningTask) { log.info(" ===== 开始微调任务 ===== stopFinetuning"); - String modelName = fineTuningTask.getJobModelName()+"-"+fineTuningTask.getDeployCount()+1; + String modelName = fineTuningTask.getJobModelName() + "-" + fineTuningTask.getDeployCount() + 1; try { log.info("开始调用HTTP服务开始微调任务,任务模型名称: {}", modelName); createTuning(fineTuningTask);