DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[update] 停用/删除基础模型校验调优添加条件

Browse Source
This commit is contained in:
Liuyang 2025-01-20 16:59:42 +08:00
parent 10aa79a62e
commit ee5befa800
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java
View File

@ -211,7 +211,7 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
List<FineTuningTaskDO> fineTuningTaskDOS = fineTuningTaskMapper.selectList(new LambdaQueryWrapper<FineTuningTaskDO>()
.eq(FineTuningTaskDO::getBaseModelId, baseModelId)
.eq(FineTuningTaskDO::getStatus, FinetuningTaskStatusEnum.WAITING.getStatus()));
.in(FineTuningTaskDO::getStatus, FinetuningTaskStatusEnum.WAITING.getStatus(),FinetuningTaskStatusEnum.TRAINING.getStatus()));
if (CollectionUtils.isNotEmpty(fineTuningTaskDOS)){
return fineTuningTaskDOS.stream().collect(Collectors.toMap(FineTuningTaskDO::getId, FineTuningTaskDO::getTaskName));