This commit is contained in:
somebody
2023-04-15 13:17:20 -05:00
parent 4dcf570407
commit d3a73aaeba
2 changed files with 2 additions and 2 deletions

View File

@@ -1916,7 +1916,7 @@ def load_model(use_gpu=True, gpu_layers=None, disk_layers=None, initial_load=Fal
model = BasicAPIInferenceModel()
elif koboldai_vars.model == "API":
from modeling.inference_models.api import APIInferenceModel
model = APIInferenceModel()
model = APIInferenceModel(url.replace("/request", ""))
elif koboldai_vars.model == "CLUSTER":
from modeling.inference_models.horde import HordeInferenceModel
model = HordeInferenceModel()