diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go index acc624d5..d718e8cb 100644 --- a/internal/storeLink/modelarts.go +++ b/internal/storeLink/modelarts.go @@ -812,10 +812,37 @@ func (m *ModelArtsLink) CheckModelExistence(ctx context.Context, name string, mt ModelName: name, ModelType: mtype, } - err := m.GetModelId(ctx, ifoption) + err := m.CheckImageExist(ctx, ifoption) if err != nil { return false } return true } + +func (m *ModelArtsLink) CheckImageExist(ctx context.Context, option *option.InferOption) error { + req := &modelarts.ListImagesReq{ + Limit: m.pageSize, + Offset: m.pageIndex, + } + ListImageResp, err := m.modelArtsRpc.ListImages(ctx, req) + if err != nil { + return err + } + var modelName string + if ListImageResp.Code == 200 { + //return errors.New("failed to get ModelId") + for _, ListImage := range ListImageResp.Data { + if option.ModelName == "ChatGLM-6B" { + modelName = "chatglm-6b" + } else { + modelName = option.ModelName + } + + if ListImage.Name == modelName { + return nil + } + } + } + return errors.New("failed to find Image ") +}