rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix api

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
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()

2
modeling/inference_models/api.py
View File

@@ -23,7 +23,7 @@ class APIException(Exception):
class APIInferenceModel(InferenceModel):
def __init__(self, base_url: str = "http://localhost:5000") -> None:
def __init__(self, base_url: str) -> None:
super().__init__()
self.base_url = base_url