From f06000a526ca096b4f72688872f52540bb74eb34 Mon Sep 17 00:00:00 2001 From: tzwang Date: Fri, 11 Oct 2024 11:10:08 +0800 Subject: [PATCH] fix shuguang instance bug --- internal/storeLink/octopus.go | 5 ++--- internal/storeLink/shuguangai.go | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/internal/storeLink/octopus.go b/internal/storeLink/octopus.go index 42157c30..a7702878 100644 --- a/internal/storeLink/octopus.go +++ b/internal/storeLink/octopus.go @@ -84,10 +84,9 @@ var ( CardModelNameCmdMap = map[string]map[string]string{ BIV100: {"blip-image-captioning-base": "pip install -U transformers; pip install fastapi uvicorn[standard]; pip install python-multipart; cd /code; python infer_biv100.py", "imagenet_resnet50": "pip install -U transformers; pip install fastapi uvicorn[standard]; pip install python-multipart; cd /code/infer; python infer_biv100.py", - "ChatGLM_6B": "su root; pip install transformers==4.33.2; pip install fastapi uvicorn[standard]; cd /code; python infer_biv100.py"}, + "ChatGLM-6B": "su root; pip install transformers==4.33.2; pip install fastapi uvicorn[standard]; cd /code; python infer_biv100.py"}, MLU: {"blip-image-captioning-base": "", - "imagenet_resnet50": "su root; . /torch/venv3/pytorch/bin/activate; pip install fastapi uvicorn[standard]; pip install python-multipart; cd /code/infer; python infer_mlu.py", - "ChatGLM_6B": ""}, + "imagenet_resnet50": "su root; . /torch/venv3/pytorch/bin/activate; pip install fastapi uvicorn[standard]; pip install python-multipart; cd /code/infer; python infer_mlu.py"}, } ) diff --git a/internal/storeLink/shuguangai.go b/internal/storeLink/shuguangai.go index 1fb447a8..e83713d0 100644 --- a/internal/storeLink/shuguangai.go +++ b/internal/storeLink/shuguangai.go @@ -879,7 +879,7 @@ func (s *ShuguangAi) GetInferDeployInstance(ctx context.Context, id string) (*in Id: id, } resp, err := s.aCRpc.GetInstanceServiceDetail(ctx, req) - if err != nil || resp.Code != "0" { + if err != nil || resp.Code != "0" || resp.Data == nil { return nil, err }