Commit 38637a6a authored by myhloli's avatar myhloli
Browse files

feat: allow additional keyword arguments in model loading function

parent 3ac969ce
......@@ -25,6 +25,7 @@ class ModelSingleton:
backend: str,
model_path: str | None,
server_url: str | None,
**kwargs,
) -> BasePredictor:
key = (backend, model_path, server_url)
if key not in self._models:
......@@ -34,6 +35,7 @@ class ModelSingleton:
backend=backend,
model_path=model_path,
server_url=server_url,
**kwargs,
)
return self._models[key]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment