Gitea
3 years ago
1 changed files with 8 additions and 7 deletions
Unified View
Diff Options
+ 8
- 7
routers/repo/modelarts.go
View File
| @@ -220,17 +220,18 @@ func Notebook2Create(ctx *context.Context, form auth.CreateModelArtsNotebookForm | |||||
| } | } | ||||
| var datasetInfos map[string]models.DatasetInfo | var datasetInfos map[string]models.DatasetInfo | ||||
| var attachSize int | |||||
| if uuid != "" { | if uuid != "" { | ||||
| datasetInfos, _, err = models.GetDatasetInfo(uuid) | datasetInfos, _, err = models.GetDatasetInfo(uuid) | ||||
| for _, infos := range datasetInfos { | for _, infos := range datasetInfos { | ||||
| if infos.Size > int(setting.DebugAttachSize * 1024 * 1024 * 1024) { | |||||
| log.Error("The DatasetSize exceeds the limit (%d)", int(setting.DebugAttachSize)) //GB | |||||
| notebookNewDataPrepare(ctx) | |||||
| ctx.RenderWithErr(ctx.Tr("cloudbrain.error.debug_datasetsize"), tplModelArtsNotebookNew, &form) | |||||
| return | |||||
| } | |||||
| attachSize += infos.Size | |||||
| } | |||||
| if attachSize > int(setting.DebugAttachSize * 1024 * 1024 * 1024) { | |||||
| log.Error("The DatasetSize exceeds the limit (%d)", int(setting.DebugAttachSize)) //GB | |||||
| notebookNewDataPrepare(ctx) | |||||
| ctx.RenderWithErr(ctx.Tr("cloudbrain.error.debug_datasetsize"), tplModelArtsNotebookNew, &form) | |||||
| return | |||||
| } | } | ||||
| } | } | ||||
| var aiCenterCode = models.AICenterOfCloudBrainTwo | var aiCenterCode = models.AICenterOfCloudBrainTwo | ||||