Set primary device to CPU if in CPU-only mode

2022-06-20 16:25:01 -04:00 · 2022-06-20 16:25:01 -04:00 · ab5ab79003
parent bd7d7b41a1
commit ab5ab79003
1 changed files with 2 additions and 0 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -616,6 +616,8 @@ def device_config(config):

    if(not breakmodel.gpu_blocks):
        print("Nothing assigned to a GPU, reverting to CPU only mode")
+        import breakmodel
+        breakmodel.primary_device = "cpu"
        vars.breakmodel = False
        vars.usegpu = False
        return