GPU detection bugfix

2021-08-20 12:30:52 +02:00 · 2021-08-20 12:30:52 +02:00 · dd77ac2f3a
parent 99c5ff240c
commit dd77ac2f3a
1 changed files with 1 additions and 1 deletions
--- a/aiserver.py
+++ b/aiserver.py
@ -198,9 +198,9 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly"]):
    elif(vars.hascuda):
        print("{0}Use GPU or CPU for generation?:  (Default GPU){1}\n".format(colors.CYAN, colors.END))
        print("    1 - GPU\n    2 - CPU\n")
+        genselected = False

    if(vars.hascuda):
-        genselected = False
        while(genselected == False):
            genselect = input("Mode> ")
            if(genselect == ""):