Merge pull request #220 from ebolam/united

Fix for loading models on CPU only that don't support breakmodel
2022-09-28 00:46:37 +02:00 · 2022-09-28 00:46:37 +02:00 · d2ff32be32
parent 057ddb4fb2 168ae8083c
commit d2ff32be32
1 changed files with 1 additions and 0 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -1516,6 +1516,7 @@ def get_model_info(model, directory=""):
        layer_count = get_layer_count(model, directory=directory)
        if layer_count is None:
            breakmodel = False
+            gpu = True
        else:
            breakmodel = True
            if model in ["NeoCustom", "GPT2Custom"]: