1

server/common/localai.go

@@ -67,12 +67,17 @@ func LocalAiClient(ctx context.Context) *LocalAi {
 	}
 }
 
-// QwenVL 调 local-ai 的多模态 VLM（qwen3.6-35b-a3b + mmproj，llama.cpp 后端）做图片推理。
-// 注意：VL 与 z-image 生成不能同时驻留 12G 显存（OOM），调用方须保证阶段互斥。
+// QwenVL 调多模态 VLM 做图片推理（OpenAI 兼容 /v1/chat/completions）。
+// 2026-09-10 起默认走 omlx（localAi.vlmBaseUrl，MLX 多模态模型）；未配置 vlmBaseUrl
+// 时回落 localAi.baseUrl（旧 local-ai 部署同机混部形态）。
 func QwenVL(ctx context.Context, imgData []byte, mime, prompt string) (string, error) {
-	base := strings.TrimRight(g.Cfg().MustGet(ctx, "localAi.baseUrl").String(), "/")
+	base := g.Cfg().MustGet(ctx, "localAi.vlmBaseUrl").String()
 	if base == "" {
-		return "", fmt.Errorf("localAi.baseUrl 未配置")
+		base = g.Cfg().MustGet(ctx, "localAi.baseUrl").String()
+	}
+	base = strings.TrimRight(base, "/")
+	if base == "" {
+		return "", fmt.Errorf("localAi.vlmBaseUrl/localAi.baseUrl 未配置")
 	}
 	model := g.Cfg().MustGet(ctx, "localAi.vlmModel", "qwen3.6-35b-a3b").String()
 	b64 := base64.StdEncoding.EncodeToString(imgData)
@@ -96,6 +101,9 @@ func QwenVL(ctx context.Context, imgData []byte, mime, prompt string) (string, e
 		return "", err
 	}
 	req.Header.Set("Content-Type", "application/json")
+	if key := g.Cfg().MustGet(ctx, "localAi.vlmApiKey").String(); key != "" {
+		req.Header.Set("Authorization", "Bearer "+key)
+	}
 	resp, err := http.DefaultClient.Do(req)
 	if err != nil {
 		return "", err

server/common/training_runner.go

@@ -42,8 +42,10 @@ type TrainingRunner interface {
 	// PushArtifact 把服务器本地文件推到训练机（增量训练基座权重，2026-09-09；
 	// remoteName 相对训练机 workdir）
 	PushArtifact(ctx context.Context, job *TrainingJob, remoteName, localPath string) error
-	// SyncYoloDataset 把训练集包落到训练机（subprocess 直写 workdir；ssh 走 tar 流式管道，本地不落盘）
+	// SyncYoloDataset 把训练集包落到训练机（subprocess 直写 workdir，原图硬链接引用；ssh 走 tar 流式管道，本地不落盘）
 	SyncYoloDataset(ctx context.Context, job *TrainingJob, pkg *YoloPackage) error
+	// CleanupYoloDataset 删除训练机上的数据集目录（任务终态调用：成功/失败/取消，防 GB 级残留）
+	CleanupYoloDataset(ctx context.Context, job *TrainingJob) error
 	// WriteTaskJson 把任务参数文件写到训练机（随 Start 前的准备阶段调用）
 	WriteTaskJson(ctx context.Context, job *TrainingJob, content string) error
 }
@@ -106,7 +108,7 @@ func TrainingConfigOf(ctx context.Context) (TrainingConfig, bool) {
 		Workdir:     g.Cfg().MustGet(ctx, "training.workdir").String(),
 		DatasetDir:  g.Cfg().MustGet(ctx, "training.datasetDir", "datasets").String(),
 		Python:      g.Cfg().MustGet(ctx, "training.venvPython").String(),
-		TimeoutMins: g.Cfg().MustGet(ctx, "training.timeoutMinutes", 600).Int(),
+		TimeoutMins: g.Cfg().MustGet(ctx, "training.timeoutMinutes", 0).Int(),
 		Model:       g.Cfg().MustGet(ctx, "training.model", "yolov8s.pt").String(),
 		Imgsz:       g.Cfg().MustGet(ctx, "training.imgsz", 1280).Int(),
 		ModelN:      g.Cfg().MustGet(ctx, "training.modelN").String(),
@@ -235,25 +237,41 @@ func (r *subprocessRunner) PushArtifact(ctx context.Context, job *TrainingJob, r
 }
 
 func (r *subprocessRunner) SyncYoloDataset(ctx context.Context, job *TrainingJob, pkg *YoloPackage) error {
-	// 同机训练：训练进程直接读 workdir 下文件，包直写目标目录（无中间暂存）
+	// 同机训练：训练进程直接读 workdir 下文件，包直写目标目录（无中间暂存）。
+	// 原图用硬链接引用项目数据集目录（同卷 0 空间占用，训练机=开发机架构下原图只存一份），
+	// 跨卷等 os.Link 失败场景回退拷贝
 	dst := filepath.Join(job.Workdir, job.DatasetDir, "yolo", job.DatasetName)
 	_ = os.RemoveAll(dst)
 	for _, f := range pkg.Files {
-		data := f.Content
+		dstPath := filepath.Join(dst, filepath.FromSlash(f.Name))
 		if f.ImagePath != "" {
+			if err := os.MkdirAll(filepath.Dir(dstPath), 0o755); err != nil {
+				return err
+			}
+			if err := os.Link(f.ImagePath, dstPath); err == nil {
+				continue
+			}
 			imgData, rErr := os.ReadFile(f.ImagePath)
 			if rErr != nil {
 				return gerror.Wrapf(rErr, "读取原图失败: %s", f.ImagePath)
 			}
-			data = imgData
+			if err := WriteFileAtomic(dstPath, imgData); err != nil {
+				return err
+			}
+			continue
 		}
-		if err := WriteFileAtomic(filepath.Join(dst, filepath.FromSlash(f.Name)), data); err != nil {
+		if err := WriteFileAtomic(dstPath, f.Content); err != nil {
 			return err
 		}
 	}
 	return nil
 }
 
+func (r *subprocessRunner) CleanupYoloDataset(ctx context.Context, job *TrainingJob) error {
+	dst := filepath.Join(job.Workdir, job.DatasetDir, "yolo", job.DatasetName)
+	return os.RemoveAll(dst)
+}
+
 func (r *subprocessRunner) WriteTaskJson(ctx context.Context, job *TrainingJob, content string) error {
 	dir := filepath.Join(job.Workdir, "tasks")
 	if err := os.MkdirAll(dir, 0o755); err != nil {
@@ -493,6 +511,12 @@ func writeYoloTar(w io.Writer, datasetName string, pkg *YoloPackage) error {
 	return nil
 }
 
+func (r *sshRunner) CleanupYoloDataset(ctx context.Context, job *TrainingJob) error {
+	dst := fmt.Sprintf("%s/%s/yolo/%s", job.Workdir, job.DatasetDir, job.DatasetName)
+	_, err := r.runCmd(ctx, job, fmt.Sprintf("rm -rf %s", dst))
+	return err
+}
+
 func (r *sshRunner) WriteTaskJson(ctx context.Context, job *TrainingJob, content string) error {
 	// 通过 ssh stdin 管道写文件（避免命令行转义地狱）
 	client, err := r.dial(ctx)