rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Set primary device to CPU if in CPU-only mode

Browse Source
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
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