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-23 18:03:33 +08:00
parent 2a119c13e3
commit 3f2ac2e513
3 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -38,6 +38,7 @@ public class AsyncFineTuningTaskService {
Long datasetId = fineTuningTask.getDataset();
DatasetRespVO dataset = datasetService.getDataset(datasetId);
req.setFileId(dataset.getJobId());
req.setSuffix(String.valueOf(fineTuningTask.getId()));
AigcFineTuningCreateRespVO resp = trainHttpService.finetuningCreate(new HashMap<>(), req);
FineTuningTaskDO updateObj = new FineTuningTaskDO();
updateObj.setId(fineTuningTask.getId());

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

@ -23,7 +23,7 @@ public class DatasetTaskSyncService {
@Resource
private AsyncDataSetService dataSetService;
// @Scheduled(cron ="0/15 * * * * ?")
@Scheduled(cron ="0/15 * * * * ?")
public void syncDatasetAigcTask() {
log.info("[syncDatasetAigcTask][开始同步任务]");
// 查询已经标注完成的数据集

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

@ -83,11 +83,13 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
public void deleteFineTuningTask(Long id) {
// 校验存在
validateFineTuningTaskExists(id);
FineTuningTaskDO fineTuningTaskDO = fineTuningTaskMapper.selectById(id);
// 删除
LambdaUpdateWrapper<FineTuningTaskDO> wrapper = new LambdaUpdateWrapper<>();
wrapper.eq(FineTuningTaskDO::getId, id)
.set(FineTuningTaskDO::getDeleted,true);
fineTuningTaskMapper.update(null,wrapper);
asyncFineTuningTaskService.stopFineTuningTask(fineTuningTaskDO);
// fineTuningTaskMapper.deleteById(id);
}