KoboldAI API model name fix

2025-06-05 21:59:24 +02:00 · 2023-05-26 11:42:29 -04:00
parent 52f5d87906
commit 0376ab5715
1 changed files with 1 additions and 0 deletions
--- a/modeling/inference_models/api/class.py
+++ b/modeling/inference_models/api/class.py
@@ -28,6 +28,7 @@ class model_backend(InferenceModel):
    def __init__(self) -> None:
        super().__init__()
        self.base_url = ""
+        self.model_name = "KoboldAI API"

    def is_valid(self, model_name, model_path, menu_path):
        return model_name == "API"