Skip to content

GitLab

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

feat: allow additional keyword arguments in model loading function

parent 3ac969ce
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
mineru/backend/vlm/vlm_analyze.py
View file @ 38637a6a
...@@ -25,6 +25,7 @@ class ModelSingleton: ...@@ -25,6 +25,7 @@ class ModelSingleton:
backend: str, backend: str,
model_path: str | None, model_path: str | None,
server_url: str | None, server_url: str | None,
**kwargs,
) -> BasePredictor: ) -> BasePredictor:
key = (backend, model_path, server_url) key = (backend, model_path, server_url)
if key not in self._models: if key not in self._models:
...@@ -34,6 +35,7 @@ class ModelSingleton: ...@@ -34,6 +35,7 @@ class ModelSingleton:
backend=backend, backend=backend,
model_path=model_path, model_path=model_path,
server_url=server_url, server_url=server_url,
**kwargs,
) )
return self._models[key] 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