From ab5ab7900375bc10234f4d7b5f2e5b0590611f61 Mon Sep 17 00:00:00 2001
From: Gnome Ann <>
Date: Mon, 20 Jun 2022 16:25:01 -0400
Subject: [PATCH] Set primary device to CPU if in CPU-only mode

---
 aiserver.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/aiserver.py b/aiserver.py
index 46d96e18..6355ba4e 100644
--- 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