mirror of
https://gitee.com/ascend/ModelLink.git
synced 2024-12-02 03:48:24 +08:00
13 lines
515 B
Bash
13 lines
515 B
Bash
# 请按照您的真实环境修改 set_env.sh 路径
|
||
source /usr/local/Ascend/ascend-toolkit/set_env.sh
|
||
|
||
|
||
python convert_ckpt.py --model-type GPT \
|
||
--load-model-type mg \
|
||
--save-model-type hf \
|
||
--model-type-hf baichuan \
|
||
--load-dir ./model_weights/Baichuan-legacy/ \
|
||
--target-tensor-parallel-size 1 \
|
||
--target-pipeline-parallel-size 1 \
|
||
--w-pack True \
|
||
--save-dir ./model_from_hf/Baichuan-hf/ # <-- 需要填入原始HF模型路径，新权重会存于./model_from_hf/Baichuan-hf/mg2hg/ |