From 9998aeb0f9607f8ff20e0f5e590d403bec34cc1b Mon Sep 17 00:00:00 2001
From: Gitea <gitea@fake.local>
Date: Mon, 28 Nov 2022 09:26:45 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E8=AE=AD=E7=BB=83=E4=BB=BB?=
 =?UTF-8?q?=E5=8A=A1=E7=A3=81=E7=9B=98=E6=8C=82=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/cloudbrain.go           | 10 ++++++----
 modules/modelarts/modelarts.go | 17 +++++++++++++++++
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index cdd9698fe..cb2bd2837 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -1045,6 +1045,7 @@ type CreateUserImageTrainJobParams struct {
 	Description string          `json:"job_desc"`
 	Config      UserImageConfig `json:"config"`
 	WorkspaceID string          `json:"workspace_id"`
+	Volume      Volumes         `json:"volumes"`
 }
 
 type UserImageConfig struct {
@@ -1063,10 +1064,11 @@ type UserImageConfig struct {
 }
 
 type CreateTrainJobParams struct {
-	JobName     string `json:"job_name"`
-	Description string `json:"job_desc"`
-	Config      Config `json:"config"`
-	WorkspaceID string `json:"workspace_id"`
+	JobName     string  `json:"job_name"`
+	Description string  `json:"job_desc"`
+	Config      Config  `json:"config"`
+	WorkspaceID string  `json:"workspace_id"`
+	Volume      Volumes `json:"volumes"`
 }
 
 type Config struct {
diff --git a/modules/modelarts/modelarts.go b/modules/modelarts/modelarts.go
index dd502dfd0..06d0836e2 100755
--- a/modules/modelarts/modelarts.go
+++ b/modules/modelarts/modelarts.go
@@ -66,6 +66,9 @@ const (
 	SortByCreateTime  = "create_time"
 	ConfigTypeCustom  = "custom"
 	TotalVersionCount = 1
+
+	VolumeSourcePath = "192.168.0.30:/"
+	VolumeDestPath   = "/home/ma-user/sfs"
 )
 
 var (
@@ -374,6 +377,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 				UserImageUrl: req.UserImageUrl,
 				UserCommand:  req.UserCommand,
 			},
+			Volume: models.Volumes{
+				Nfs: models.Nfs{
+					SourcePath: VolumeSourcePath,
+					DestPath:   VolumeDestPath,
+					ReadOnly:   false,
+				},
+			},
 		})
 	} else {
 		jobResult, createErr = createTrainJob(models.CreateTrainJobParams{
@@ -394,6 +404,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 				},
 				Parameter: req.Parameters,
 			},
+			Volume: models.Volumes{
+				Nfs: models.Nfs{
+					SourcePath: VolumeSourcePath,
+					DestPath:   VolumeDestPath,
+					ReadOnly:   false,
+				},
+			},
 		})
 	}
 	if createErr != nil {