模型调优

This commit is contained in:
sunxiqing 2025-01-03 12:19:11 +08:00
parent 04ae52ec6e
commit eea8bd3fc6

View File

@ -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());