1 changed files with 5 additions and 5 deletions
Unified View
Diff Options
+ 5
- 5
routers/repo/modelarts.go
View File
| @@ -119,7 +119,7 @@ func MustEnableModelArts(ctx *context.Context) { | |||||
| func NotebookNew(ctx *context.Context) { | func NotebookNew(ctx *context.Context) { | ||||
| notebookNewDataPrepare(ctx) | notebookNewDataPrepare(ctx) | ||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "", models.JobTypeDebug) | |||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "") | |||||
| ctx.Data["WaitCount"] = waitCount | ctx.Data["WaitCount"] = waitCount | ||||
| ctx.HTML(200, tplModelArtsNotebookNew) | ctx.HTML(200, tplModelArtsNotebookNew) | ||||
| } | } | ||||
| @@ -631,7 +631,7 @@ func TrainJobNew(ctx *context.Context) { | |||||
| ctx.ServerError("get new train-job info failed", err) | ctx.ServerError("get new train-job info failed", err) | ||||
| return | return | ||||
| } | } | ||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "", models.JobTypeTrain) | |||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "") | |||||
| ctx.Data["WaitCount"] = waitCount | ctx.Data["WaitCount"] = waitCount | ||||
| ctx.HTML(200, tplModelArtsTrainJobNew) | ctx.HTML(200, tplModelArtsTrainJobNew) | ||||
| } | } | ||||
| @@ -785,7 +785,7 @@ func TrainJobNewVersion(ctx *context.Context) { | |||||
| ctx.ServerError("get new train-job info failed", err) | ctx.ServerError("get new train-job info failed", err) | ||||
| return | return | ||||
| } | } | ||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "", models.JobTypeTrain) | |||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "") | |||||
| ctx.Data["WaitCount"] = waitCount | ctx.Data["WaitCount"] = waitCount | ||||
| ctx.HTML(200, tplModelArtsTrainJobVersionNew) | ctx.HTML(200, tplModelArtsTrainJobVersionNew) | ||||
| } | } | ||||
| @@ -2057,7 +2057,7 @@ func InferenceJobIndex(ctx *context.Context) { | |||||
| PageSize: setting.UI.IssuePagingNum, | PageSize: setting.UI.IssuePagingNum, | ||||
| }, | }, | ||||
| RepoID: repo.ID, | RepoID: repo.ID, | ||||
| Type: models.TypeCloudBrainAll, | |||||
| Type: ctx.QueryInt("type"), | |||||
| JobTypes: jobTypes, | JobTypes: jobTypes, | ||||
| }) | }) | ||||
| if err != nil { | if err != nil { | ||||
| @@ -2100,7 +2100,7 @@ func InferenceJobNew(ctx *context.Context) { | |||||
| ctx.ServerError("get new inference-job info failed", err) | ctx.ServerError("get new inference-job info failed", err) | ||||
| return | return | ||||
| } | } | ||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "", models.JobTypeInference) | |||||
| waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "") | |||||
| ctx.Data["WaitCount"] = waitCount | ctx.Data["WaitCount"] = waitCount | ||||
| ctx.HTML(200, tplModelArtsInferenceJobNew) | ctx.HTML(200, tplModelArtsInferenceJobNew) | ||||
| } | } | ||||