From 66aed1b0a5ec80e8434369e891124a07a40771e4 Mon Sep 17 00:00:00 2001 From: limin Date: Thu, 9 Jan 2025 17:31:21 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../service/finetuningtask/FineTuningTaskServiceImpl.java | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java index aaea244c3..d5e76977a 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskServiceImpl.java @@ -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);