rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

GPU detection bugfix

This commit is contained in:
henk717 2021-08-20 12:30:52 +02:00
parent 99c5ff240c
commit dd77ac2f3a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -198,9 +198,9 @@ if(not vars.model in ["InferKit", "Colab", "OAI", "ReadOnly"]):
elif(vars.hascuda): elif(vars.hascuda):
print("{0}Use GPU or CPU for generation?: (Default GPU){1}\n".format(colors.CYAN, colors.END)) print("{0}Use GPU or CPU for generation?: (Default GPU){1}\n".format(colors.CYAN, colors.END))
print(" 1 - GPU\n 2 - CPU\n") print(" 1 - GPU\n 2 - CPU\n")
genselected = False
if(vars.hascuda): if(vars.hascuda):
genselected = False
while(genselected == False): while(genselected == False):
genselect = input("Mode> ") genselect = input("Mode> ")
if(genselect == ""): if(genselect == ""):