[QUESTION]How to convert the weight file format of the MAMBA model from pt to safetensors format?

[fxnie]

Ask a clear and concise question about Megatron-LM.
MODEL_PATH="/workspace/mnt/xxx/ckpt/mamba2"
SAVE_PATH="/workspace/mnt/xxx/models/convert_ckpt/mamba2"

python tools/checkpoint/hybrid_conversion.py
--load-dir ${MODEL_PATH}
--save-dir ${SAVE_PATH}
--target-tp-size 2
--target-pp-size 1
--d-model 2560
--mamba-version 2
--mamba-d-state 128
--mamba2-n-groups 8
--mamba2-head-dim 64

I found that the conversion script is still in pt format after conversion. What should I do to convert it to SafeTensor format? help

[github-actions]

Marking as stale. No activity in 60 days.

[github-actions]

This issue was closed because it has been inactive for 7 days since being marked as stale.