diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go
index 557e7f9a5..a88da8fe5 100644
--- a/models/ai_model_manage.go
+++ b/models/ai_model_manage.go
@@ -374,6 +374,12 @@ func QueryModel(opts *AiModelQueryOptions) ([]*AiModelManage, int64, error) {
 		)
 	}
 
+	if (opts.Status) >= 0 {
+		cond = cond.And(
+			builder.Eq{"ai_model_manage.status": opts.Status},
+		)
+	}
+
 	count, err := sess.Where(cond).Count(new(AiModelManage))
 	if err != nil {
 		return nil, 0, fmt.Errorf("Count: %v", err)
diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 324377c0e..18ebfc050 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -851,6 +851,7 @@ func QueryModelListForPredict(ctx *context.Context) {
 		RepoID: repoId,
 		Type:   ctx.QueryInt("type"),
 		New:    -1,
+		Status: 0,
 	})
 	if err != nil {
 		ctx.ServerError("Cloudbrain", err)