test/ModelLink2
1
0
Fork 0
mirror of https://gitee.com/ascend/ModelLink.git synced 2024-11-29 18:38:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

!1798 Aquila2权重转换mg-hf切换新框架

Browse Source 
Merge pull request !1798 from qyz/master
This commit is contained in:
qyz 2024-10-31 06:36:48 +00:00 committed by i-robot
parent 3e9fd1d561
commit 44715d4fec
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/legacy/aquila2/ckpt_convert_aquila2_legacy2hf.sh
View File

@ -3,9 +3,9 @@ source /usr/local/Ascend/ascend-toolkit/set_env.sh
# --params-dtype bf16 结合需要使用
python convert_ckpt.py --model-type GPT \
--loader megatron \
--saver megatron \
--save-model-type save_huggingface_llama \
--load-model-type mg \
--save-model-type hf \
--model-type-hf llama2 \
--load-dir ./model_weights/Aquila2-legacy/ \
--target-tensor-parallel-size 1 \
--target-pipeline-parallel-size 1 \