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 04fb4412c..ff53ba001 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 @@ -92,7 +92,7 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService { DatasetDO datasetDO = datasetMapper.selectById(dataset); result.setDatasetName(datasetDO.getDatasetName()); } - Long baseModelId = Long.valueOf(fineTuningTaskDO.getBaseModel()); + Long baseModelId = fineTuningTaskDO.getBaseModelId(); if (baseModelId != null) { BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId); result.setBaseModelName(baseModelDO.getModelName());