floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'updated deployinstancelist logics' (#303) from tzwang/pcm-coordinator:master into master 

Former-commit-id: 6b1f3a9b6a2606cd448994089d84620c0c8ca107
This commit is contained in:
tzwang 2024-09-09 09:31:27 +08:00
parent 67f000865a 19838aab83
commit 985f2ef6e5
3 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/logic/inference/deployinstancelistlogic.go
View File

@ -38,6 +38,14 @@ func (l *DeployInstanceListLogic) DeployInstanceList(req *types.DeployInstanceLi
return nil, tx.Error
}
if len(tasklist) == 0 {
resp.List = nil
resp.PageSize = req.PageSize
resp.PageNum = req.PageNum
resp.Total = 0
return
}
//count total
var total int64
err = tx.Count(&total).Error

6
internal/logic/inference/getdeploytasksbytypelogic.go
View File

@ -70,12 +70,6 @@ func removeItem(items *[]*models.AiDeployInstanceTask, id int64) {
if len(*items) == 0 {
return
}
if len(*items) == 1 {
if (*items)[0].Id == id {
(*items) = nil
return
}
}
for i := len(*items) - 1; i >= 0; i-- {
if (*items)[i].Id == id {
*items = append((*items)[:i], (*items)[i+1:]...)

2
internal/storeLink/storeLink.go
View File

@ -80,7 +80,7 @@ var (
}
ModelTypeMap = map[string][]string{
"image_classification": {"imagenet_resnet50"},
"text_to_text": {"chatGLM_6B"},
"text_to_text": {"ChatGLM-6B"},
"image_to_text": {"blip-image-captioning-base"},
"text_to_image": {"stable-diffusion-xl-base-1.0"},
}