File size: 360 Bytes
a86f2f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# coding=utf-8
# Copyright (c) 2025 Huawei Technologies Co., Ltd. All rights reserved.
rm -rf ./model
mkdir ./model
python split_weight.py \
--model_path=../ \
--output_path=./model \
--origin_yaml_file_path=./runner_config/tp1.yaml \
--new_yaml_file_path=./runner_config/tp32.yaml \
--world_size=32 \
--node_num=1 \
--node_rank=0
|