Skip to content

M
mogo
Commit 08aca3ed authored by vicotor's avatar vicotor
Browse files
Options

fix bug

parent 68564477
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
operator/workerinfo.go
View file @ 08aca3ed
......@@ -218,9 +218,9 @@ func (d *WorkerInfoOperator) FindWorkerByInstallModelAndSortByGpuRam(ctx context
// sort by gpu ram
findOptions := options.Find()
findOptions.SetLimit(int64(limit))
findOptions.SetSort(bson.D{{"hardware.gpu.ram", -1}})
findOptions.SetSort(bson.D{{"hardware.GPU.ram", -1}})
selector := bson.M{"model_infos.installed_models.model_id": modelId, "hardware.gpu.performance": bson.M{"$gte": performance}, "hardware.gpu.ram": bson.M{"$gte": ram}}
selector := bson.M{"model_infos.installed_models.model_id": modelId, "hardware.GPU.performance": bson.M{"$gte": performance}, "hardware.GPU.ram": bson.M{"$gte": ram}}
cursor, err := d.col.Find(ctx, selector, findOptions)
if err != nil {
return nil, err
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment