no gpu if vram < 2GB
This commit is contained in:
parent
d890890f66
commit
811c3d1900
|
@ -215,6 +215,11 @@ func CheckVRAM() (int64, error) {
|
|||
free += vram
|
||||
}
|
||||
|
||||
if free*1024*1024 < 2*1000*1000*1000 {
|
||||
log.Printf("less than 2 GB VRAM available, falling back to CPU only")
|
||||
free = 0
|
||||
}
|
||||
|
||||
return free, nil
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in a new issue