添加训练任务磁盘挂载

models/cloudbrain.go

@@ -1045,7 +1045,6 @@ type CreateUserImageTrainJobParams struct {
 	Description string          `json:"job_desc"`
 	Config      UserImageConfig `json:"config"`
 	WorkspaceID string          `json:"workspace_id"`
-	DataMount   Nfs             `json:"volumes"`
 }
 
 type UserImageConfig struct {
@@ -1061,6 +1060,7 @@ type UserImageConfig struct {
 	CreateVersion bool        `json:"create_version"`
 	Flavor        Flavor      `json:"flavor"`
 	PoolID        string      `json:"pool_id"`
+	DataMount     Volumes     `json:"volumes"`
 }
 
 type CreateTrainJobParams struct {
@@ -1068,7 +1068,6 @@ type CreateTrainJobParams struct {
 	Description string `json:"job_desc"`
 	Config      Config `json:"config"`
 	WorkspaceID string `json:"workspace_id"`
-	DataMount   Nfs    `json:"volumes"`
 }
 
 type Config struct {
@@ -1082,9 +1081,10 @@ type Config struct {
 	LogUrl        string      `json:"log_url"`
 	//UserImageUrl 		string       	`json:"user_image_url"`
 	//UserCommand  		string 			`json:"user_command"`
-	CreateVersion bool   `json:"create_version"`
-	Flavor        Flavor `json:"flavor"`
-	PoolID        string `json:"pool_id"`
+	CreateVersion bool    `json:"create_version"`
+	Flavor        Flavor  `json:"flavor"`
+	PoolID        string  `json:"pool_id"`
+	DataMount     Volumes `json:"volumes"`
 }
 type CreateInferenceJobParams struct {
 	JobName     string    `json:"job_name"`

modules/modelarts/modelarts.go

@@ -376,11 +376,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 				Parameter:    req.Parameters,
 				UserImageUrl: req.UserImageUrl,
 				UserCommand:  req.UserCommand,
-			},
-			DataMount: models.Nfs{
-				SourcePath: VolumeSourcePath,
-				DestPath:   VolumeDestPath,
-				ReadOnly:   false,
+				DataMount: models.Volumes{
+					Nfs: models.Nfs{
+						SourcePath: VolumeSourcePath,
+						DestPath:   VolumeDestPath,
+						ReadOnly:   false,
+					},
+				},
 			},
 		})
 	} else {
@@ -401,11 +403,13 @@ func GenerateTrainJob(ctx *context.Context, req *GenerateTrainJobReq) (jobId str
 					Code: req.Spec.SourceSpecId,
 				},
 				Parameter: req.Parameters,
-			},
-			DataMount: models.Nfs{
-				SourcePath: VolumeSourcePath,
-				DestPath:   VolumeDestPath,
-				ReadOnly:   false,
+				DataMount: models.Volumes{
+					Nfs: models.Nfs{
+						SourcePath: VolumeSourcePath,
+						DestPath:   VolumeDestPath,
+						ReadOnly:   false,
+					},
+				},
 			},
 		})
 	}