| @@ -778,6 +778,10 @@ datasets = Datasets | |||||
| datasets.desc = Enable Dataset | datasets.desc = Enable Dataset | ||||
| cloudbrain_helper=Use GPU/NPU resources to open notebooks, model training tasks, etc. | cloudbrain_helper=Use GPU/NPU resources to open notebooks, model training tasks, etc. | ||||
| model_manager = Model | |||||
| model_noright=No right | |||||
| model_rename=Duplicate model name, please modify model name. | |||||
| debug=Debug | debug=Debug | ||||
| stop=Stop | stop=Stop | ||||
| delete=Delete | delete=Delete | ||||
| @@ -784,6 +784,7 @@ cloudbrain_helper=使用GPU/NPU资源,开启Notebook、模型训练任务等 | |||||
| model_manager = 模型 | model_manager = 模型 | ||||
| model_noright=无权限操作 | model_noright=无权限操作 | ||||
| model_rename=模型名称重复,请修改模型名称 | |||||
| debug=调试 | debug=调试 | ||||
| stop=停止 | stop=停止 | ||||
| @@ -105,6 +105,23 @@ func saveModelByParameters(jobId string, versionName string, name string, versio | |||||
| return nil | return nil | ||||
| } | } | ||||
| func SaveNewNameModel(ctx *context.Context) { | |||||
| name := ctx.Query("Name") | |||||
| if name == "" { | |||||
| ctx.Error(500, fmt.Sprintf("name or version is null.")) | |||||
| return | |||||
| } | |||||
| aimodels := models.QueryModelByName(name, ctx.Repo.Repository.ID) | |||||
| if len(aimodels) > 0 { | |||||
| ctx.Error(500, ctx.Tr("repo.model_rename")) | |||||
| return | |||||
| } | |||||
| SaveModel(ctx) | |||||
| log.Info("save model end.") | |||||
| } | |||||
| func SaveModel(ctx *context.Context) { | func SaveModel(ctx *context.Context) { | ||||
| log.Info("save model start.") | log.Info("save model start.") | ||||
| JobId := ctx.Query("JobId") | JobId := ctx.Query("JobId") | ||||
| @@ -977,6 +977,7 @@ func RegisterRoutes(m *macaron.Macaron) { | |||||
| }, context.RepoRef()) | }, context.RepoRef()) | ||||
| m.Group("/modelmanage", func() { | m.Group("/modelmanage", func() { | ||||
| m.Post("/create_model", reqRepoModelManageWriter, repo.SaveModel) | m.Post("/create_model", reqRepoModelManageWriter, repo.SaveModel) | ||||
| m.Post("/create_new_model", reqRepoModelManageWriter, repo.SaveNewNameModel) | |||||
| m.Delete("/delete_model", repo.DeleteModel) | m.Delete("/delete_model", repo.DeleteModel) | ||||
| m.Put("/modify_model", repo.ModifyModelInfo) | m.Put("/modify_model", repo.ModifyModelInfo) | ||||
| m.Get("/show_model", reqRepoModelManageReader, repo.ShowModelTemplate) | m.Get("/show_model", reqRepoModelManageReader, repo.ShowModelTemplate) | ||||