diff --git a/modules/auth/cloudbrain.go b/modules/auth/cloudbrain.go index 91fa8ad64..160328b5b 100755 --- a/modules/auth/cloudbrain.go +++ b/modules/auth/cloudbrain.go @@ -69,6 +69,7 @@ type CreateCloudBrainInferencForm struct { ModelName string `form:"model_name" binding:"Required"` ModelVersion string `form:"model_version" binding:"Required"` CkptName string `form:"ckpt_name" binding:"Required"` + LabelName string `form:"label_names" binding:"Required"` } func (f *CreateCloudBrainForm) Validate(ctx *macaron.Context, errs binding.Errors) binding.Errors { diff --git a/modules/cloudbrain/cloudbrain.go b/modules/cloudbrain/cloudbrain.go index dbfff6027..6cbb97999 100755 --- a/modules/cloudbrain/cloudbrain.go +++ b/modules/cloudbrain/cloudbrain.go @@ -78,6 +78,7 @@ type GenerateCloudBrainTaskReq struct { ModelName string ModelVersion string CkptName string + LabelName string } func GetCloudbrainDebugCommand() string { @@ -400,6 +401,7 @@ func GenerateTask(req GenerateCloudBrainTaskReq) error { ModelVersion: req.ModelVersion, CkptName: req.CkptName, ResultUrl: req.ResultPath, + LabelName: req.LabelName, CreatedUnix: createTime, UpdatedUnix: createTime, CommitID: req.CommitID, diff --git a/routers/api/v1/repo/cloudbrain.go b/routers/api/v1/repo/cloudbrain.go index 5ebbc9b46..d291024f9 100755 --- a/routers/api/v1/repo/cloudbrain.go +++ b/routers/api/v1/repo/cloudbrain.go @@ -157,15 +157,15 @@ func DelCloudBrainJob(ctx *context.APIContext) { if errStr != "" { ctx.JSON(http.StatusOK, map[string]interface{}{ - "message": ctx.Tr(errStr), + "Message": ctx.Tr(errStr), "VersionName": "1", - "code": 1, + "Code": 1, }) } else { ctx.JSON(http.StatusOK, map[string]interface{}{ - "message": "", + "Message": "", "VersionName": "1", - "code": 0, + "Code": 0, }) } diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index 20a1fcc07..5c03e26a7 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -380,6 +380,7 @@ func CloudBrainInferenceJobCreate(ctx *context.Context, form auth.CreateCloudBra codePath := setting.JobPath + jobName + cloudbrain.CodeMountPath resourceSpecId := form.ResourceSpecId branchName := form.BranchName + labelName := form.LabelName repo := ctx.Repo.Repository ckptUrl := setting.Attachment.Minio.RealPath + form.TrainUrl + form.CkptName @@ -475,6 +476,7 @@ func CloudBrainInferenceJobCreate(ctx *context.Context, form auth.CreateCloudBra ModelVersion: form.ModelVersion, CkptName: form.CkptName, TrainUrl: form.TrainUrl, + LabelName: labelName, } err = cloudbrain.GenerateTask(req) @@ -830,6 +832,8 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo } ctx.Data["task"] = task + labelName := strings.Fields(task.LabelName) + ctx.Data["LabelName"] = labelName ctx.Data["jobName"] = task.JobName ctx.Data["displayJobName"] = task.DisplayJobName version_list_task := make([]*models.Cloudbrain, 0) diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl index 7653376dd..909cde468 100644 --- a/templates/repo/cloudbrain/inference/show.tmpl +++ b/templates/repo/cloudbrain/inference/show.tmpl @@ -266,7 +266,7 @@ onclick="javascript:parseInfo()">{{$.i18n.Tr "repo.cloudbrain.runinfo"}} {{$.i18n.Tr "repo.model_download"}} + data-gpu-flag="true" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}