diff --git a/README-zh-Hans.md b/README-zh-Hans.md index f94d639..1f05812 100644 --- a/README-zh-Hans.md +++ b/README-zh-Hans.md @@ -142,7 +142,7 @@ streamlit run web_demo.py 通过 InternLM 进行训练的模型可以很轻松地转换为 HuggingFace Transformers 格式,方便与社区各种开源项目无缝对接。借助 `tools/transformers/convert2hf.py` 可以将训练保存的权重一键转换为 transformers 格式 ```bash -python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer .tools/V7_sft.model +python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer ./tools/V7_sft.model ``` 转换之后可以通过以下的代码加载为 transformers diff --git a/README.md b/README.md index 056bb82..fcf9c0f 100644 --- a/README.md +++ b/README.md @@ -145,7 +145,7 @@ Please refer to [Usage Tutorial](./doc/en/usage.md) to start InternLM installati The model trained by InternLM can be easily converted to HuggingFace Transformers format, which is convenient for seamless docking with various open source projects in the community. With the help of `tools/transformers/convert2hf.py`, the weights saved during training can be converted into transformers format with one command ```bash -python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer .tools/V7_sft.model +python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer ./tools/V7_sft.model ``` After conversion, it can be loaded as transformers by the following code diff --git a/tools/transformers/README-zh-Hans.md b/tools/transformers/README-zh-Hans.md index 8f14692..7337424 100644 --- a/tools/transformers/README-zh-Hans.md +++ b/tools/transformers/README-zh-Hans.md @@ -11,7 +11,7 @@ `convert2hf.py` 可以将训练保存的权重一键转换为 transformers 格式。在仓库根目录运行以下命令: ```bash -python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer .tools/V7_sft.model +python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer ./tools/V7_sft.model ``` 然后可以使用 `from_pretrained` 接口加载: diff --git a/tools/transformers/README.md b/tools/transformers/README.md index 901eeb3..0c4ecc8 100644 --- a/tools/transformers/README.md +++ b/tools/transformers/README.md @@ -10,7 +10,7 @@ This folder contains the `InternLM` model in transformers format. `convert2hf.py` can convert saved training weights into the transformers format with a single command. Execute the command in the root directory of repository: ```bash -python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer .tools/V7_sft.model +python tools/transformers/convert2hf.py --src_folder origin_ckpt/ --tgt_folder hf_ckpt/ --tokenizer ./tools/V7_sft.model ``` Then, you can load it using the `from_pretrained` interface: