From 92ebac652a94c5e4f449388e59d4ee7c6c17842d Mon Sep 17 00:00:00 2001 From: tzwang Date: Tue, 15 Oct 2024 16:10:03 +0800 Subject: [PATCH] fix shuguangai inference cmd --- internal/storeLink/shuguangai.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/internal/storeLink/shuguangai.go b/internal/storeLink/shuguangai.go index e83713d0..412072d9 100644 --- a/internal/storeLink/shuguangai.go +++ b/internal/storeLink/shuguangai.go @@ -45,8 +45,8 @@ const ( WorkPath = "/work/home/acgnnmfbwo/pcmv1/" TimeoutLimit = "10:00:00" PythonCodePath = "/work/home/acgnnmfbwo/111111/py/test.py" - DATASETS_DIR = "/work/home/acgnnmfbwo/pcmv1/dataset" - ALGORITHM_DIR = "/work/home/acgnnmfbwo/pcmv1/algorithm" + DATASETS_DIR = KUNSHAN_DIR + "/dataset" + ALGORITHM_DIR = KUNSHAN_DIR + "/algorithm" KUNSHAN_DIR = "/public/home/acgnnmfbwo/pcmv1" TRAIN_FILE = "train.py" CPUCOREPRICEPERHOUR = 0.09 @@ -97,8 +97,8 @@ var ( } ModelNameCmdMap = map[string]string{ - "blip-image-captioning-base": "pip install transformers python-multipart fastapi uvicorn[standard]; python /public/home/acgnnmfbwo/pcmv1/inference/pytorch/blip_image_captioning_base/infer.py", - "imagenet_resnet50": "pip install fastapi uvicorn[standard] python-multipart; python /public/home/acgnnmfbwo/pcmv1/inference/pytorch/imagenet_resnet50/infer.py", + "blip-image-captioning-base": "sudo pip install transformers python-multipart fastapi uvicorn[standard]; sudo python /public/home/acgnnmfbwo/pcmv1/inference/pytorch/blip_image_captioning_base/infer.py", + "imagenet_resnet50": "sudo pip install fastapi uvicorn[standard] python-multipart; sudo python /public/home/acgnnmfbwo/pcmv1/inference/pytorch/imagenet_resnet50/infer.py", } ) @@ -883,6 +883,10 @@ func (s *ShuguangAi) GetInferDeployInstance(ctx context.Context, id string) (*in return nil, err } + if resp.Data == nil { + return nil, errors.New("GetInferDeployInstance empty") + } + var url string if resp.Data.Status == constants.Running { url = resp.Data.ContainerPortInfoList[0].AccessUrl