Deeper disable of --nobreakmodel attempt

2022-08-11 17:47:19 -04:00 · 2022-08-11 17:47:19 -04:00 · bddcd7ab7f
parent 45495d8792
commit bddcd7ab7f
1 changed files with 4 additions and 0 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -545,6 +545,10 @@ def device_config(config):
    n_layers = utils.num_layers(config)
    if args.cpu:
        breakmodel.gpu_blocks = [0]*n_layers
+        return
+    elif vars.nobreakmodel:
+        breakmodel.gpu_blocks = [0]*n_layers
+        return
    elif(args.breakmodel_gpulayers is not None or (utils.HAS_ACCELERATE and args.breakmodel_disklayers is not None)):
        try:
            if(not args.breakmodel_gpulayers):