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:30:21 +08:00
parent 4127a80428
commit cd33cef810
7 changed files with 20 additions and 0 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/framework/backend/config/LLMBackendProperties.java
View File

@ -74,6 +74,7 @@ public class LLMBackendProperties {
private String finetuningDetail;
private String finetuningFileList;
private String finetuningCancel;
private String applicationApi;

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

@ -85,6 +85,7 @@ public class AsyncFineTuningTaskService {
@Async
public void stopFineTuningTask(FineTuningTaskDO fineTuningTask) {
try {
trainHttpService.finetuningCancle(new HashMap<>(),fineTuningTask.getJobId());
trainHttpService.finetuningDelete(new HashMap<>(), fineTuningTask.getJobId());
}catch(Exception e){
e.printStackTrace();

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

@ -62,6 +62,7 @@ public class FineTuningTaskServiceImpl implements FineTuningTaskService {
// 插入
validateFineTuningNameTaskExists(createReqVO);
FineTuningTaskDO fineTuningTask = BeanUtils.toBean(createReqVO, FineTuningTaskDO.class);
fineTuningTask.setStatus(FinetuningTaskStatusEnum.WAITING.getStatus());
fineTuningTaskMapper.insert(fineTuningTask);
//todo 调用模型服务，创建调优任务
asyncFineTuningTaskService.createTuning(fineTuningTask);

11
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
View File

@ -312,4 +312,15 @@ public class TrainHttpService {
}
public AigcRespVO finetuningCancle(Map<String, String> headers, String jobId) {
login(headers);
String finetuningCancel = llmBackendProperties.getFinetuningCancel();
String url = String.format(finetuningCancel,jobId);
String res = HttpUtils.del(url, headers);
log.info(" finetuningDelete:{}", res);
AigcRespVO aigcRespVO = JSON.parseObject(res, AigcRespVO.class);
log.info(" finetuningDelete:{}", aigcRespVO);
return aigcRespVO;
}
}

2
yudao-server/src/main/resources/application-dev.yaml
View File

@ -243,6 +243,8 @@ llm:
finetuning_create: http://36.103.199.104:9000/api/finetuning
# 微调任务详情 GET
finetuning_detail: http://36.103.199.104:9000/api/finetuning
# 微调任务取消
finetuning_cancel: http://36.103.199.104:9000/api/finetuning/%s/cancel
# 微调文件列表 GET
finetuning_file_list: http://36.103.199.104:9000/api/files?purpose=fine-tune
# 模型创建

2
yudao-server/src/main/resources/application-local.yaml
View File

@ -286,6 +286,8 @@ llm:
finetuning_create: http://36.103.199.104:9000/api/finetuning
# 微调任务详情 GET
finetuning_detail: http://36.103.199.104:9000/api/finetuning
# 微调任务取消
finetuning_cancel: http://36.103.199.104:9000/api/finetuning/%s/cancel
# 微调文件列表 GET
finetuning_file_list: http://36.103.199.104:9000/api/files?purpose=fine-tune
# 模型部署

2
yudao-server/src/main/resources/application-prod.yaml
View File

@ -286,6 +286,8 @@ llm:
finetuning_create: http://36.133.1.230:8080/api/finetuning
# 微调任务详情 GET
finetuning_detail: http://36.133.1.230:8080/api/finetuning
# 微调任务取消
finetuning_cancel: http://36.103.199.104:9000/api/finetuning/%s/cancel
# 微调文件列表 GET
finetuning_file_list: http://36.133.1.230:8080/api/files?purpose=fine-tune
# 模型部署