routers/repo/ai_model_manage.go

@@ -1193,6 +1193,30 @@ func isCanDownload(ctx *context.Context, task *models.AiModelManage) bool {
 	return false
 }
 
+func isModelCanDownload(user *models.User, repo *models.Repository, task *models.AiModelManage) bool {
+	if user == nil {
+		return false
+	}
+	isCollaborator, err := repo.IsCollaborator(user.ID)
+	if err != nil {
+		log.Info("query error.")
+	}
+	isTeamMember, err := repo.IsInRepoTeam(user.ID)
+	if err != nil {
+		log.Info("query IsInRepoTeam error." + err.Error())
+	}
+	if user.IsAdmin || user.ID == task.UserId || isCollaborator || isTeamMember {
+		return true
+	}
+	if repo.OwnerID == user.ID {
+		return true
+	}
+	if !task.IsPrivate {
+		return true
+	}
+	return false
+}
+
 func isQueryPrivateModel(ctx *context.Context) bool {
 	if ctx.User == nil {
 		return false

routers/repo/ai_model_square.go

@@ -189,6 +189,7 @@ func ModelSquareData(ctx *context.Context) {
 				model.RepoOwnerName = repo.OwnerName
 				model.RepoDisplayName = repo.DisplayName()
 			}
+			model.IsCanDownload = isModelCanDownload(ctx.User, repo, model)
 		}
 		if ctx.User != nil && modelCollect != nil {
 			value := modelCollect[model.ID]

web_src/vuepages/components/cloudbrain/ModelSelectV2.vue

@@ -210,7 +210,7 @@ export default {
         dataI._modelName = dataI.name;
         dataI._modelVersion = dataI.version;
         dataI._preTrainModelUrl = dataI.path;
-        if (this.useExceedSize && dataI.size > this.exceedSize && this.exceedSize) {
+        if (this.useExceedSize && dataI.size > this.exceedSize && this.exceedSize || !dataI.isCanDownload) {
           dataI.disabled = true;
         }
         _children.forEach(item => {