From 181c93424c92a1073f21486b86d792cd5732ce62 Mon Sep 17 00:00:00 2001 From: ebolam Date: Tue, 30 Aug 2022 15:10:11 -0400 Subject: [PATCH] Fix for KoboldAI API as a model option --- aiserver.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/aiserver.py b/aiserver.py index 2ce2afe5..bf4cfac7 100644 --- a/aiserver.py +++ b/aiserver.py @@ -1426,7 +1426,9 @@ def get_model_info(model, directory=""): gpu_names = [] for i in range(gpu_count): gpu_names.append(torch.cuda.get_device_name(i)) - if model in [x[1] for x in model_menu['apilist']]: + if model in ['Colab', 'API']: + url = True + elif model in [x[1] for x in model_menu['apilist']]: if path.exists("settings/{}.settings".format(model)): with open("settings/{}.settings".format(model), "r") as file: # Check if API key exists @@ -1439,8 +1441,6 @@ def get_model_info(model, directory=""): key = True elif model == 'ReadOnly': pass - elif model == 'Colab': - url = True elif not utils.HAS_ACCELERATE and not torch.cuda.is_available(): pass elif args.cpu: