dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-23 19:24:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 14

feat: use a metadata file
Some checks failed
Security Scan / tests (push) Has been cancelled
Details

Browse source 
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2025-06-20 22:18:48 +02:00
parent 9a34fc8c66
commit ec137888ba
7 changed files with 164 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

23
core/system/capabilities.go
View file

@ -25,15 +25,22 @@ func detectGPUVendor() (string, error) {
}
for _, gpu := range gpus {
if strings.ToUpper(gpu.DeviceInfo.Vendor.Name) == "NVIDIA" {
return "nvidia", nil
}
if strings.ToUpper(gpu.DeviceInfo.Vendor.Name) == "AMD" {
return "amd", nil
}
if strings.ToUpper(gpu.DeviceInfo.Vendor.Name) == "INTEL" {
return "intel", nil
if gpu.DeviceInfo != nil {
if gpu.DeviceInfo.Vendor != nil {
gpuVendorName := strings.ToUpper(gpu.DeviceInfo.Vendor.Name)
if gpuVendorName == "NVIDIA" {
return "nvidia", nil
}
if gpuVendorName == "AMD" {
return "amd", nil
}
if gpuVendorName == "INTEL" {
return "intel", nil
}
return "nvidia", nil
}
}
}
return "", nil