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

fix getadepterByModel bugs 

Former-commit-id: 1dc4ccd87fc22d7df3ebc427caede9aa69a82a0b
This commit is contained in:
tzwang 2024-10-24 16:14:29 +08:00
parent 1bbb35d3cf
commit 92428bcc06
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
internal/logic/inference/getadaptersbymodellogic.go
View File

@ -38,7 +38,19 @@ func (l *GetAdaptersByModelLogic) GetAdaptersByModel(req *types.GetAdaptersByMod
}
for _, cluster := range clusters.List {
exist := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id][cluster.Id].CheckModelExistence(l.ctx, req.ModelName, req.ModelType)
cmap, found := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id]
if !found {
continue
}
iCluster, found := cmap[cluster.Id]
if !found {
continue
}
exist := iCluster.CheckModelExistence(l.ctx, req.ModelName, req.ModelType)
if exist {
c := &types.ClusterAvail{
ClusterId: cluster.Id,