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-24 19:59:58 +08:00
parent 04005e5f95
commit f8feca0247
4 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/modelassesstaskmanual/ModelAssessTaskManualMapper.java
View File

@ -9,6 +9,7 @@ import cn.iocoder.yudao.module.llm.dal.dataobject.modelassesstaskmanual.ModelAss
import org.apache.ibatis.annotations.Mapper;
import cn.iocoder.yudao.module.llm.controller.admin.modelassesstaskmanual.vo.*;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;
import org.apache.ibatis.annotations.Update;
/**
@ -27,4 +28,7 @@ public interface ModelAssessTaskManualMapper extends BaseMapperX<ModelAssessTask
}
@Update("update llm_model_assess_task_manual set status = #{status}, progress = #{formattedRatio} where id = #{modelAnswerId}")
void updateStatus(@Param("modelAnswerId") Long modelAnswerId,@Param("formattedRatio") Integer formattedRatio,@Param("status") Integer status);
@Select("select backup_id from llm_model_assess_task_manual where id = #{id}")
Long selectTrueById(Long id);
}

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/async/AsyncFineTuningTaskService.java
View File

@ -79,7 +79,7 @@ public class AsyncFineTuningTaskService {
req.setProcPerNode(fineTuningTask.getGpuCount());
req.setLearningRate(fineTuningTask.getLearningRate());
req.setModelMaxLength(fineTuningTask.getCutoffLen());
// req.setLora(fineTuningTask.getLorayRank() != null);
req.setLora(true);
return req;
}

2
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/vo/AigcFineTuningCreateReqVO.java
View File

@ -22,5 +22,5 @@ public class AigcFineTuningCreateReqVO {
private int procPerNode;
private double learningRate;
private int modelMaxLength;
// private boolean lora;
private boolean lora;
}

12
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/modelassesstaskmanual/ModelAssessTaskManualServiceImpl.java
View File

@ -263,9 +263,15 @@ public class ModelAssessTaskManualServiceImpl implements ModelAssessTaskManualSe
@Override
public void unBackup (Long id) {
ModelAssessTaskManualDO modelAssessTaskManualDO = modelAssessTaskManualMapper.selectById(id);
this.modelAssessTaskManualBackupService.unBackup(modelAssessTaskManualDO.getBackupId());
modelAssessTaskManualDO.setBackupId(0L);
modelAssessTaskManualMapper.updateById(modelAssessTaskManualDO);
Long backupId;
if (modelAssessTaskManualDO == null){
backupId = modelAssessTaskManualMapper.selectTrueById(id);
}else {
backupId = modelAssessTaskManualDO.getBackupId();
modelAssessTaskManualDO.setBackupId(0L);
modelAssessTaskManualMapper.updateById(modelAssessTaskManualDO);
}
this.modelAssessTaskManualBackupService.unBackup(backupId);
}
/**