浏览代码

add

tags/v1.22.12.1^2
Gitea 3 年前
父节点
当前提交
8545f920d3
共有 2 个文件被更改,包括 12 次插入0 次删除
  1. +6
    -0
      models/cloudbrain.go
  2. +6
    -0
      modules/modelarts/modelarts.go

+ 6
- 0
models/cloudbrain.go 查看文件

@@ -1155,6 +1155,9 @@ type TrainJobVersionConfig struct {
Flavor Flavor `json:"flavor"` Flavor Flavor `json:"flavor"`
PoolID string `json:"pool_id"` PoolID string `json:"pool_id"`
PreVersionId int64 `json:"pre_version_id"` PreVersionId int64 `json:"pre_version_id"`
ShareAddr string `json:"nas_share_addr"`
MountPath string `json:"nas_mount_path"`
NasType string `json:"nas_type"`
} }


type TrainJobVersionUserImageConfig struct { type TrainJobVersionUserImageConfig struct {
@@ -1170,6 +1173,9 @@ type TrainJobVersionUserImageConfig struct {
PreVersionId int64 `json:"pre_version_id"` PreVersionId int64 `json:"pre_version_id"`
UserImageUrl string `json:"user_image_url"` UserImageUrl string `json:"user_image_url"`
UserCommand string `json:"user_command"` UserCommand string `json:"user_command"`
ShareAddr string `json:"nas_share_addr"`
MountPath string `json:"nas_mount_path"`
NasType string `json:"nas_type"`
} }


type CreateConfigParams struct { type CreateConfigParams struct {


+ 6
- 0
modules/modelarts/modelarts.go 查看文件

@@ -521,6 +521,9 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, job
PreVersionId: req.PreVersionId, PreVersionId: req.PreVersionId,
UserImageUrl: req.UserImageUrl, UserImageUrl: req.UserImageUrl,
UserCommand: req.UserCommand, UserCommand: req.UserCommand,
ShareAddr: ShareAddr,
MountPath: MountPath,
NasType: NasType,
}, },
}, jobId) }, jobId)
} else { } else {
@@ -540,6 +543,9 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, job
}, },
Parameter: req.Parameters, Parameter: req.Parameters,
PreVersionId: req.PreVersionId, PreVersionId: req.PreVersionId,
ShareAddr: ShareAddr,
MountPath: MountPath,
NasType: NasType,
}, },
}, jobId) }, jobId)
} }


正在加载...
取消
保存