diff --git a/tools/load_internlm_model.py b/tools/load_internlm_model.py
index 123cade..1824939 100644
--- a/tools/load_internlm_model.py
+++ b/tools/load_internlm_model.py
@@ -134,7 +134,7 @@ def initialize_internlm_model(
     """Initialize internlm model.
 
     Args:
-        model_type (str): The types of models supported by train_internlm, such as "INTERNLM".
+        model_type (str): The types of models supported by internlm framework, such as "INTERNLM".
         ckpt_dir (str): Directory where model checkpoints are stored. Its format needs to be like this:
             (a) local path, such as: "local:{your local path}";
             (b) boto3 path, such as: "boto3:s3://{bucket name}.{ip}/{your ceph path}".