From ecd23ba389e9abcf37ce2f5a0b2d5adfd5833a3a Mon Sep 17 00:00:00 2001 From: tzwang Date: Mon, 30 Oct 2023 10:48:46 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AD=98=E7=AE=97=E8=81=94=E5=8A=A8=E4=BF=AE?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Former-commit-id: e6e7389ce22dd1928bfe759cdba7613b662524cd --- api/internal/storeLink/modelarts.go | 2 +- api/internal/storeLink/storeLink.go | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/api/internal/storeLink/modelarts.go b/api/internal/storeLink/modelarts.go index 66ea5e66..dd22f4da 100644 --- a/api/internal/storeLink/modelarts.go +++ b/api/internal/storeLink/modelarts.go @@ -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) { diff --git a/api/internal/storeLink/storeLink.go b/api/internal/storeLink/storeLink.go index b8b49e2e..24cde5bb 100644 --- a/api/internal/storeLink/storeLink.go +++ b/api/internal/storeLink/storeLink.go @@ -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, } )