fix-bug

models/attachment.go

@@ -61,6 +61,7 @@ type AttachmentUsername struct {
 type AttachmentInfo struct {
 	Attachment `xorm:"extends"`
 	Repo       *Repository `xorm:"extends"`
+	User       *User       `xorm:"extends"`
 }
 
 type AttachmentsOptions struct {
@@ -612,6 +613,12 @@ func Attachments(opts *AttachmentsOptions) ([]*AttachmentInfo, int64, error) {
 			} else {
 				return nil, 0, fmt.Errorf("GetRepositoryByID failed error: %v", err)
 			}
+			user, err := GetUserByID(attachment.UploaderID)
+			if err == nil {
+				attachment.User = user
+			} else {
+				return nil, 0, fmt.Errorf("GetUserByID failed error: %v", err)
+			}
 		}
 	}
 

routers/repo/cloudbrain.go

@@ -162,6 +162,8 @@ func cloudBrainNewDataPrepare(ctx *context.Context) error {
 	ctx.Data["brainscore_path"] = cloudbrain.BrainScoreMountPath
 	ctx.Data["is_brainscore_enabled"] = setting.IsBrainScoreEnabled
 
+	ctx.Data["cloudbraintype"] = models.TypeCloudBrainOne
+
 	return nil
 }
 

routers/repo/modelarts.go

@@ -132,6 +132,8 @@ func notebookNewDataPrepare(ctx *context.Context) error {
 	}
 	ctx.Data["flavors"] = modelarts.FlavorInfos.FlavorInfo
 
+	ctx.Data["cloudbraintype"] = models.TypeCloudBrainTwo
+
 	return nil
 }
 
@@ -578,6 +580,7 @@ func trainJobNewDataPrepare(ctx *context.Context) error {
 		return err
 	}
 	ctx.Data["config_list"] = configList.ParaConfigs
+	ctx.Data["cloudbraintype"] = models.TypeCloudBrainTwo
 
 	return nil
 }
@@ -751,6 +754,7 @@ func trainJobNewVersionDataPrepare(ctx *context.Context) error {
 	ctx.Data["uuid"] = task.Uuid
 	ctx.Data["flavor_code"] = task.FlavorCode
 	ctx.Data["engine_id"] = task.EngineID
+	ctx.Data["cloudbraintype"] = models.TypeCloudBrainTwo
 
 	configList, err := getConfigList(modelarts.PerPage, 1, modelarts.SortByCreateTime, "desc", "", modelarts.ConfigTypeCustom)
 	if err != nil {
@@ -1955,6 +1959,7 @@ func inferenceJobNewDataPrepare(ctx *context.Context) error {
 		New:    MODEL_LATEST,
 	})
 	ctx.Data["MODEL_COUNT"] = model_count
+	ctx.Data["cloudbraintype"] = models.TypeCloudBrainTwo
 
 	return nil
 }