From a058258fc0985ec85bd755e64b5b00bcef9cd90b Mon Sep 17 00:00:00 2001
From: YWMditto <862779238@qq.com>
Date: Thu, 9 Nov 2023 17:01:50 +0800
Subject: [PATCH] fix some info

---
 tools/load_internlm_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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}".