提交代码

modules/cloudbrain/cloudbrain.go

@@ -256,13 +256,7 @@ func GenerateTask(req GenerateCloudBrainTaskReq) error {
 				ReadOnly:  false,
 			},
 		},
-		{
-			HostPath: models.StHostPath{
-				Path:      req.ModelPath,
-				MountPath: ModelMountPath,
-				ReadOnly:  false,
-			},
-		},
+
 		{
 			HostPath: models.StHostPath{
 				Path:      req.BenchmarkPath,
@@ -292,6 +286,25 @@ func GenerateTask(req GenerateCloudBrainTaskReq) error {
 			},
 		},
 	}
+	if len(req.CkptName) == 0 {
+		volumes = append(volumes, models.Volume{
+			HostPath: models.StHostPath{
+				Path:      req.ModelPath,
+				MountPath: ModelMountPath,
+				ReadOnly:  false,
+			},
+		})
+
+	} else {
+		volumes = append(volumes, models.Volume{
+			HostPath: models.StHostPath{
+				Path:      req.ModelPath,
+				MountPath: ModelMountPath + "/" + req.CkptName,
+				ReadOnly:  false,
+			},
+		})
+
+	}
 
 	if len(req.DatasetInfos) == 1 {
 		volumes = append(volumes, models.Volume{