提交代码

routers/repo/cloudbrain.go

@@ -2769,7 +2769,7 @@ func getTrainJobCommand(form auth.CreateCloudBrainForm) (string, error) {
 		}
 	}
 	if form.CkptName != "" {
-		param += " --pretrainmodelname" + "=" + form.CkptName
+		param += " --ckpt_url" + "=" + "/pretrainmodel/" + form.CkptName
 	}
 
 	command += "python /code/" + bootFile + param + " > " + cloudbrain.ModelMountPath + "/" + form.DisplayJobName + "-" + cloudbrain.LogFile

routers/repo/grampus.go

@@ -994,7 +994,7 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo
 	}
 
 	if pretrainModelFileName != "" {
-		paramCode += " --pretrainmodelname" + "=" + pretrainModelFileName
+		paramCode += " --ckpt_url" + "=" + workDir + "pretrainmodel/" + pretrainModelFileName
 	}
 
 	var commandCode string

routers/repo/modelarts.go

@@ -1054,7 +1054,7 @@ func TrainJobCreate(ctx *context.Context, form auth.CreateModelArtsTrainJobForm)
 	}
 	if !account.IsPointBalanceEnough(ctx.User.ID, spec.UnitPrice*form.WorkServerNumber) {
 		log.Error("point balance is not enough,userId=%d specId=%d", ctx.User.ID, spec.ID)
-		cloudBrainNewDataPrepare(ctx)
+		trainJobNewDataPrepare(ctx)
 		ctx.RenderWithErr(ctx.Tr("points.insufficient_points_balance"), tplModelArtsTrainJobNew, &form)
 		return
 	}