From b500785333aa3ef174afdb01b83e4ab082f560ab Mon Sep 17 00:00:00 2001 From: liuzx Date: Thu, 6 Jan 2022 09:54:52 +0800 Subject: [PATCH] update --- routers/repo/modelarts.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index 76236f00e..bbe0d177b 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -1826,6 +1826,19 @@ func inferenceJobNewDataPrepare(ctx *context.Context) error { } ctx.Data["config_list"] = configList.ParaConfigs + repoId := ctx.Repo.Repository.ID + Type := -1 + _, model_count, _ := models.QueryModel(&models.AiModelQueryOptions{ + ListOptions: models.ListOptions{ + Page: 1, + PageSize: 2, + }, + RepoID: repoId, + Type: Type, + New: MODEL_LATEST, + }) + ctx.Data["MODEL_COUNT"] = model_count + return nil }