rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Set primary device to CPU if in CPU-only mode

This commit is contained in:
Gnome Ann 2022-06-20 16:25:01 -04:00
parent bd7d7b41a1
commit ab5ab79003
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -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