mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-04-25 01:19:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

docs: add more information about pretrained weights

Browse Source 
Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
aarnphm-ec2-dev
2023-06-10 06:58:36 +00:00
parent c669d38dea
commit 8fbf352ec6
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/openllm/cli.py
View File

@@ -316,10 +316,12 @@ def start_model_command(
ModelEnv = openllm.utils.ModelEnv(model_name)
llm_config = openllm.AutoConfig.for_model(model_name)
for_doc = openllm.AutoLLM.for_model(model_name)
docstring = f"""\
{ModelEnv.start_docstring}
\b
The available pretrained models to use with '{model_name}' are: {openllm.AutoLLM.for_model(model_name).pretrained}
The available pretrained models to use with '{model_name}' are: {for_doc.pretrained} [default: {for_doc.default_model}]
Tip: One can pass one of the aforementioned to '--pretrained' to use other pretrained weights.
"""
command_attrs: dict[str, t.Any] = {
"name": ModelEnv.model_name,