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-02-21 10:17:32 +08:00
parent 1b25d0f767
commit c552063dea
2 changed files with 5 additions and 2 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/service/finetuningtask/FineTuningTaskServiceImpl.java
View File

@ -176,10 +176,10 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
result.setBaseModelName(baseModelDO.getModelName());
}
}
if (StringUtil.isNotEmpty(fineTuningTaskDO.getJobId())) {
/* if (StringUtil.isNotEmpty(fineTuningTaskDO.getJobId())) {
AigcFineTuningDetailRespVO aigcFineTuningDetailRespVO = trainHttpService.finetuningDetail(new HashMap<>(), fineTuningTaskDO.getJobId());
result.setDetailRespVO(aigcFineTuningDetailRespVO);
}
}*/
result.setCheckPointDataJSON(JSON.parseObject(fineTuningTaskDO.getCheckPointData()));
return result;
}

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

@ -52,6 +52,9 @@ public class FineTuningTaskSyncService {
if(Objects.equals(fineTuningTaskDO.getStatus(), FinetuningTaskStatusEnum.TRAINING.getStatus())
|| Objects.equals(fineTuningTaskDO.getStatus(), FinetuningTaskStatusEnum.WAITING.getStatus())){
ServerNameDO serverNameDO = serverNameMapper.selectById(fineTuningTaskDO.getGpuType());
if (fineTuningTaskDO.getJobId() == null){
continue;
}
String hostUrl = serverNameDO!=null ?serverNameDO.getHost():"";
String queryJobs = "?filter={\"job_id\":\""+fineTuningTaskDO.getJobId()+"\"}";
String respJobs = trainHttpService.modelTableQuery(new HashMap<>(), hostUrl,"fine_tuning_train_job",queryJobs);