Skip to content

Commit 156bd4b

Browse files
committed
feat: ensure model aliases unique
1 parent 19eff36 commit 156bd4b

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

llama_cpp/server/model.py

+6
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,12 @@ def __init__(self, models: List[ModelSettings]) -> None:
1919
for model in models:
2020
if not model.model_alias:
2121
model.model_alias = model.model
22+
if model.model_alias in self._model_settings_dict:
23+
raise ValueError(
24+
f"Please specify a unique model alias for each model: {model.model_alias}"
25+
)
26+
if model.verbose:
27+
print(f"Registering model: {model.model_alias}")
2228
self._model_settings_dict[model.model_alias] = model
2329

2430
self._current_model: Optional[llama_cpp.Llama] = None

0 commit comments

Comments
 (0)