diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index 1155f3239..205660402 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -1004,7 +1004,11 @@ func isCanDownload(ctx *context.Context, task *models.AiModelManage) bool {
 	if err != nil {
 		log.Info("query error.")
 	}
-	if ctx.User.IsAdmin || ctx.User.ID == task.UserId || isCollaborator {
+	isTeamMember, err := ctx.Repo.Repository.IsInRepoTeam(ctx.User.ID)
+	if err != nil {
+		log.Info("query IsInRepoTeam error." + err.Error())
+	}
+	if ctx.User.IsAdmin || ctx.User.ID == task.UserId || isCollaborator || isTeamMember {
 		return true
 	}
 	if ctx.Repo.IsOwner() {