From 8fbf352ec67901b59f7780cd352d866a37727975 Mon Sep 17 00:00:00 2001
From: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
Date: Sat, 10 Jun 2023 06:58:36 +0000
Subject: [PATCH] docs: add more information about pretrained weights

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
---
 src/openllm/cli.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/openllm/cli.py b/src/openllm/cli.py
index eb2db379..720df98e 100644
--- a/src/openllm/cli.py
+++ b/src/openllm/cli.py
@@ -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,