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 }