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

修复

Browse Source
This commit is contained in:
limin 2025-01-09 17:31:21 +08:00
parent 906dd7b3c4
commit 66aed1b0a5
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -114,12 +114,16 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
Long dataset = fineTuningTaskDO.getDataset();
if (dataset != null) {
DatasetDO datasetDO = datasetMapper.selectById(dataset);
result.setDatasetName(datasetDO.getDatasetName());
if (datasetDO != null){
result.setDatasetName(datasetDO.getDatasetName());
}
}
Long baseModelId = fineTuningTaskDO.getBaseModelId();
if (baseModelId != null) {
BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId);
result.setBaseModelName(baseModelDO.getModelName());
if (baseModelDO != null){
result.setBaseModelName(baseModelDO.getModelName());
}
}
AigcFineTuningDetailRespVO aigcFineTuningDetailRespVO = trainHttpService.finetuningDetail(new HashMap<>(), fineTuningTaskDO.getJobId());
result.setDetailRespVO(aigcFineTuningDetailRespVO);