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 ff53ba001..04fb4412c 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 = fineTuningTaskDO.getBaseModelId(); + Long baseModelId = Long.valueOf(fineTuningTaskDO.getBaseModel()); if (baseModelId != null) { BaseModelDO baseModelDO = baseModelMapper.selectById(baseModelId); result.setBaseModelName(baseModelDO.getModelName());