ColossalAI/examples/language/openmoe/model/convert_openmoe_ckpt.sh

2 lines
130 B
Bash

python convert_openmoe_ckpt.py --t5x_checkpoint_path /path/to/t5x --config_file /path/to/config --pytorch_dump_path /path/to/save