提交代码

modules/auth/grampus.go

@@ -31,15 +31,20 @@ func (f *CreateGrampusTrainJobForm) Validate(ctx *macaron.Context, errs binding.
 }
 
 type CreateGrampusNotebookForm struct {
-	Type           int    `form:"type"`
-	DisplayJobName string `form:"display_job_name" binding:"Required"`
-	Attachment     string `form:"attachment"`
-	ImageID        string `form:"image_id" binding:"Required"`
-	Description    string `form:"description"`
-	BranchName     string `form:"branch_name" binding:"Required"`
-	Image          string `form:"image" binding:"Required"`
-	DatasetName    string `form:"dataset_name"`
-	SpecId         int64  `form:"spec_id"  binding:"Required"`
+	Type             int    `form:"type"`
+	DisplayJobName   string `form:"display_job_name" binding:"Required"`
+	Attachment       string `form:"attachment"`
+	ImageID          string `form:"image_id" binding:"Required"`
+	Description      string `form:"description"`
+	BranchName       string `form:"branch_name" binding:"Required"`
+	Image            string `form:"image" binding:"Required"`
+	DatasetName      string `form:"dataset_name"`
+	ModelName        string `form:"model_name"`
+	ModelVersion     string `form:"model_version"`
+	CkptName         string `form:"ckpt_name"`
+	LabelName        string `form:"label_names"`
+	PreTrainModelUrl string `form:"pre_train_model_url"`
+	SpecId           int64  `form:"spec_id"  binding:"Required"`
 }
 
 func (f *CreateGrampusNotebookForm) Validate(ctx *macaron.Context, errs binding.Errors) binding.Errors {

modules/auth/modelarts.go

@@ -16,13 +16,18 @@ func (f *CreateModelArtsForm) Validate(ctx *macaron.Context, errs binding.Errors
 }
 
 type CreateModelArtsNotebookForm struct {
-	DisplayJobName string `form:"display_job_name" binding:"Required"`
-	JobName        string `form:"job_name" binding:"Required"`
-	Attachment     string `form:"attachment"`
-	Description    string `form:"description"`
-	Flavor         string `form:"flavor" binding:"Required"`
-	ImageId        string `form:"image_id" binding:"Required"`
-	SpecId         int64  `form:"spec_id" binding:"Required"`
+	DisplayJobName   string `form:"display_job_name" binding:"Required"`
+	JobName          string `form:"job_name" binding:"Required"`
+	Attachment       string `form:"attachment"`
+	Description      string `form:"description"`
+	Flavor           string `form:"flavor" binding:"Required"`
+	ImageId          string `form:"image_id" binding:"Required"`
+	ModelName        string `form:"model_name"`
+	ModelVersion     string `form:"model_version"`
+	CkptName         string `form:"ckpt_name"`
+	LabelName        string `form:"label_names"`
+	PreTrainModelUrl string `form:"pre_train_model_url"`
+	SpecId           int64  `form:"spec_id" binding:"Required"`
 }
 
 func (f *CreateModelArtsNotebookForm) Validate(ctx *macaron.Context, errs binding.Errors) binding.Errors {