conver-model.sh 360 Bytes
Newer Older
hepj's avatar
hepj committed
1
2
3
4
5
6
7
8
9
10
11
cd /public/new-pai/Pai-Megatron-Patch/toolkits/model_checkpoints_convertor/deepseek
bash hf2mcore_deepseek_v2_moe_convertor.sh \
    A2.4B \
    /public/new-pai/model/deepseek-ckpts/DeepSeek-V2-Lite \
    /public/new-pai/Pai-Megatron-Patch/examples/deepseek_v2/DeepSeek-V2-Lite-to-mcore-tp2-pp1-ep4  \
    2  \
    1  \
    4 \
    fp32 \
    true \
    false