Browse Source

存算联动修改

Former-commit-id: e6e7389ce2
pull/9/head
tzwang 2 years ago
parent
commit
ecd23ba389
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      api/internal/storeLink/modelarts.go
  2. +2
    -0
      api/internal/storeLink/storeLink.go

+ 1
- 1
api/internal/storeLink/modelarts.go View File

@@ -26,7 +26,7 @@ type ModelArtsLink struct {
//)

func NewModelArtsLink(ctx context.Context, svcCtx *svc.ServiceContext, participant *models.ScParticipantPhyInfo) *ModelArtsLink {
return &ModelArtsLink{ctx: ctx, svcCtx: svcCtx, platform: participant.Name, pageIndex: 1, pageSize: 100}
return &ModelArtsLink{ctx: ctx, svcCtx: svcCtx, participant: participant, platform: participant.Name, pageIndex: 1, pageSize: 100}
}

func (o *ModelArtsLink) UploadImage(path string) (interface{}, error) {


+ 2
- 0
api/internal/storeLink/storeLink.go View File

@@ -31,6 +31,7 @@ const (
TYPE_SHUGUANGAI = "3"
OCTOPUS = "Octopus"
MODELARTS = "Modelarts"
SHUGUANGAI = "ShuguangAi"
DCU = "dcu"
PYTORCH = "Pytorch"
TASK_PYTORCH_PREFIX = "PytorchTask"
@@ -50,6 +51,7 @@ var (
AITYPE = map[string]string{
"1": OCTOPUS,
"2": MODELARTS,
"3": SHUGUANGAI,
}
)



Loading…
Cancel
Save