github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix some info

pull/478/head
YWMditto 2023-11-09 17:01:50 +08:00
parent 8f8fe84c03
commit a058258fc0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/load_internlm_model.py
View File

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