mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Set primary device to CPU if in CPU-only mode
This commit is contained in:
@ -616,6 +616,8 @@ def device_config(config):
|
|||||||
|
|
||||||
if(not breakmodel.gpu_blocks):
|
if(not breakmodel.gpu_blocks):
|
||||||
print("Nothing assigned to a GPU, reverting to CPU only mode")
|
print("Nothing assigned to a GPU, reverting to CPU only mode")
|
||||||
|
import breakmodel
|
||||||
|
breakmodel.primary_device = "cpu"
|
||||||
vars.breakmodel = False
|
vars.breakmodel = False
|
||||||
vars.usegpu = False
|
vars.usegpu = False
|
||||||
return
|
return
|
||||||
|
Reference in New Issue
Block a user