GPU detection bugfix

This commit is contained in:
henk717 2021-08-20 12:30:52 +02:00
parent 99c5ff240c
commit dd77ac2f3a

View File

@ -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 == ""):