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

Model: Rename "Colab" to "Basic API"

Browse Source
This commit is contained in:
somebody
2023-03-07 17:07:01 -06:00
parent dc3dd38c4e
commit 6b45367cc7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modeling/inference_models/colab.py → modeling/inference_models/basic_api.py
View File

@@ -15,11 +15,11 @@ from modeling.inference_model import (
)
class ColabException(Exception):
"""To be used for errors when using the Colab API as an interface."""
class BasicAPIException(Exception):
"""To be used for errors when using the Basic API as an interface."""
class ColabInferenceModel(InferenceModel):
class BasicAPIInferenceModel(InferenceModel):
def __init__(self) -> None:
super().__init__()
@@ -65,7 +65,7 @@ class ColabInferenceModel(InferenceModel):
req = requests.post(utils.koboldai_vars.colaburl, json=reqdata)
if req.status_code != 200:
raise ColabException(f"Bad status code {req.status_code}")
raise BasicAPIException(f"Bad status code {req.status_code}")
# Deal with the response
js = req.json()["data"]