Browse Source

add

tags/v1.22.12.1^2
Gitea 3 years ago
parent
commit
8545f920d3
2 changed files with 12 additions and 0 deletions
  1. +6
    -0
      models/cloudbrain.go
  2. +6
    -0
      modules/modelarts/modelarts.go

+ 6
- 0
models/cloudbrain.go View File

@@ -1155,6 +1155,9 @@ type TrainJobVersionConfig struct {
Flavor Flavor `json:"flavor"`
PoolID string `json:"pool_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 {
@@ -1170,6 +1173,9 @@ type TrainJobVersionUserImageConfig struct {
PreVersionId int64 `json:"pre_version_id"`
UserImageUrl string `json:"user_image_url"`
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 {


+ 6
- 0
modules/modelarts/modelarts.go View File

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


Loading…
Cancel
Save