diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java
index e990ce045..a9c31d14e 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java
@@ -54,4 +54,8 @@ public class BaseModelSaveReqVO {
     private Long modelId;
 
     private String chatUrl;
+
+    private Long gpuId;
+
+    private String gpuName;
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java
index f423b8e5e..18b0b7c4d 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java
@@ -79,4 +79,8 @@ public class BaseModelDO extends BaseDO {
 
     private String chatUrl;
 
+    private Long gpuId;
+
+    private String gpuName;
+
 }
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
index 1ac5d89ee..5232cfb63 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
@@ -26,6 +26,7 @@ import cn.iocoder.yudao.module.llm.service.datarefluxdata.DataRefluxDataService;
 import cn.iocoder.yudao.module.llm.service.http.ModelService;
 import cn.iocoder.yudao.module.llm.service.http.vo.*;
 import cn.iocoder.yudao.module.llm.service.prompttemplates.PromptTemplatesService;
+import cn.iocoder.yudao.module.llm.service.servername.ServerNameService;
 import com.alibaba.excel.util.StringUtils;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
@@ -170,6 +171,7 @@ public class ConversationServiceImpl implements ConversationService {
             if (baseModelDO == null) {
                 throw exception(BASE_MODEL_NOT_EXISTS);
             }
+            selfModelUrl = baseModelDO.getChatUrl() + "/v1/chat/completions";
             model = baseModelDO.getAigcModelName();
         }else if (Objects.equals(0, chatReqVO.getModelType())) {
             // 自定义模型
diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java
index 4c7d860af..216e98663 100644
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/finetuningtask/FineTuningTaskSyncService.java
@@ -127,7 +127,7 @@ public class FineTuningTaskSyncService {
                     try {
                         //获取检查点信息
                         //todo 模型工厂的功能有问题，暂时写死
-                        jobModelName = "Qwen2.5-0.5B-Instruct-147";
+//                        jobModelName = "Qwen2.5-0.5B-Instruct-147";
                         String checkFileList = trainHttpService.getCheckFileList(hostUrl,jobModelName);
                         List<String> checkpoints = new ArrayList<>();
                         List<String> fileUrls = new ArrayList<>();