修复模型调优停止

2025-01-24 16:39:30 +08:00 · 2025-01-24 16:39:30 +08:00 · 392d4ce473
commit 392d4ce473
parent ff7a4a7e46
1 changed files with 2 additions and 0 deletions
--- 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
@ -100,6 +100,8 @@ public class AsyncFineTuningTaskService {
        try {
            trainHttpService.finetuningCancle(new HashMap<>(),fineTuningTask.getJobId());
            trainHttpService.finetuningDelete(new HashMap<>(), fineTuningTask.getJobId());
+            fineTuningTask.setStatus(0);
+            fineTuningTaskMapper.updateById(fineTuningTask);
        }catch(Exception e){
            e.printStackTrace();